Update app.py

app.py

@@ -10,13 +10,14 @@ def predict(inputs, top_p, temperature, openai_api_key, chat_counter, chatbot=[]
     response = make_request(API_URL, openai_api_key, payload)
 
     # Processar a resposta
-    chat, new_history, token_counter = process_response(response, history)
+    new_chatbot, new_history, token_counter = process_response(response, history)
 
-    # Atualizar o histórico apenas se houver novos tokens
+    # Atualizar o chatbot e o histórico apenas se houver novos tokens
     if token_counter > 0:
+        chatbot.extend(new_chatbot)
         history = new_history
 
-    return chat, history, token_counter
+    return chatbot, history, chat_counter
 
 def format_messages(chatbot, inputs, chat_counter):
     messages = []
@@ -54,7 +55,7 @@ def process_response(response, history):
     for chunk in response.iter_lines():
         if chunk:
             chunk_str = chunk.decode('utf-8').lstrip('data: ')
-            if "[DONE]" in chunk_str:
+            if chunk_str.strip() in ["[DONE]", "}"]:  # Adiciona } à verificação
                 break
             try:
                 chunk_json = json.loads(chunk_str)