Update app.py

app.py

@@ -18,50 +18,45 @@ def respond(message, history, system_message, max_tokens, temperature, top_p):
 
         # --- Passo 1: Llama 3.1 ---
         result_main = client_main.text_generation(
-            inputs=full_prompt,
-            max_tokens=max_tokens,
+            prompt=full_prompt,
+            max_new_tokens=max_tokens,
             temperature=temperature,
             top_p=top_p
         )
-        response_main = result_main[0]["generated_text"]
+        response_main = result_main.generated_text
 
         # --- Passo 2: FLAN-T5 ---
         result_aux1 = client_aux1.text_generation(
-            inputs=f"Reformule este texto de forma clara:\n{response_main}",
+            prompt=f"Reformule este texto de forma clara:\n{response_main}",
             max_new_tokens=max_tokens
         )
-        response_aux1 = result_aux1[0]["generated_text"]
+        response_aux1 = result_aux1.generated_text
 
         # --- Passo 3: BART ---
         result_aux2 = client_aux2.text_generation(
-            inputs=f"Resuma este texto em 3 frases:\n{response_aux1}",
+            prompt=f"Resuma este texto em 3 frases:\n{response_aux1}",
             max_new_tokens=150
         )
-        response_aux2 = result_aux2[0]["generated_text"]
+        response_aux2 = result_aux2.generated_text
 
     except Exception as e:
         response_aux2 = f"Erro ao gerar resposta: {e}"
 
     # Atualiza histórico no formato correto para o Gradio Chatbot
-        history.append({"role": "user", "content": message})
-        history.append({"role": "assistant", "content": response_aux2})
-
-        return history, history
-
-
-    # Atualiza histórico do chat
     history.append({"role": "user", "content": message})
     history.append({"role": "assistant", "content": response_aux2})
 
-    return response_aux2, history
+    return history, history
 
 # Interface Gradio
 with gr.Blocks() as demo:
     gr.Markdown("## 🤖 Chatbot em Cascata (Llama 3.1 + FLAN-T5 + BART)")
+
     system_message = gr.Textbox(
         value="Você é um chatbot amigável e prestativo.", 
         label="System Message"
     )
+
     chatbot = gr.Chatbot()
     msg = gr.Textbox(label="Digite sua mensagem")
     max_tokens = gr.Slider(50, 2048, 512, step=50, label="Max Tokens")