Update app.py

app.py

@@ -1,7 +1,7 @@
 import gradio as gr
 import requests
 import json
-import os 
+import os
 
 # Definir variáveis de ambiente ou substituir com sua chave de API real
 API_KEY = os.getenv('API_KEY')
@@ -25,7 +25,7 @@ def user(message, history):
     print(f"User message: {message}")
     history = history or []
     history.append({"role": "user", "content": message})
-    return "", history
+    return history
 
 def call_api(history, max_tokens, temperature, top_p, system_message="", seed=42):
     print("Calling API...")
@@ -70,7 +70,9 @@ def call_api(history, max_tokens, temperature, top_p, system_message="", seed=42
 def chat(history, system_message, max_tokens, temperature, top_p, top_k, repetition_penalty):
     print("Starting chat...")
     system_message_to_use = system_message if system_message.strip() else BASE_SYSTEM_MESSAGE
-    if history and "role" in history[-1] and history[-1]["role"] == "user":
+    if history and history[-1]["role"] == "user":
+        history = user(history[-1]["content"], history)
+    else:
         history.append({"role": "system", "content": system_message_to_use})
     assistant_response = call_api(history, max_tokens, temperature, top_p)
     if assistant_response:
@@ -94,11 +96,10 @@ with gr.Blocks() as demo:
     
     def update_chatbot(message, chat_history):
         print("Updating chatbot...")
-        chat_history, _ = user(message, chat_history) 
+        chat_history = user(message, chat_history)
         chat_history, _, _ = chat(chat_history, system_msg.value, max_tokens.value, temperature.value, top_p.value, 40, 1.1)
         return chat_history, chat_history, ""
 
-
     submit.click(
         fn=update_chatbot,
         inputs=[message, chat_history_state],
@@ -106,9 +107,9 @@ with gr.Blocks() as demo:
     )
 
     clear.click(
-        fn=lambda: clear_chat([], ""),
+        fn=clear_chat,
         inputs=[chat_history_state, message],
         outputs=[chat_history_state, message]
     )
 
-demo.launch()
+demo.launch()