Update app.py

app.py

@@ -45,15 +45,17 @@ def process_response(response, history):
     partial_words = "" 
     for chunk in response.iter_lines():
         if chunk:
-            # Removendo 'data:' do início do chunk
             chunk_str = chunk.decode('utf-8').lstrip('data: ')
+            # Verificar se a transmissão está concluída
+            if "[DONE]" in chunk_str:
+                break
             try:
                 chunk_json = json.loads(chunk_str)
-                print("Chunk JSON:", chunk_json)  # Log para diagnóstico
                 if 'choices' in chunk_json and len(chunk_json['choices']) > 0:
                     chunk_data = chunk_json['choices'][0].get('delta', {})
                     if 'content' in chunk_data:
-                        partial_words += chunk_data['content']
+                        content = chunk_data['content']
+                        partial_words += content
                         if token_counter == 0:
                             history.append(" " + partial_words)
                         else:
@@ -61,12 +63,15 @@ def process_response(response, history):
                         token_counter += 1
             except json.JSONDecodeError as e:
                 print("Error decoding JSON response:", e)
-                print("Raw chunk:", chunk_str)  # Log do chunk bruto para diagnóstico
+                print("Raw chunk:", chunk_str)
+    
+    # Finalizar a construção da resposta
     chat = [(history[i], history[i + 1]) for i in range(0, len(history) - 1, 2)]
     return chat, history, token_counter
 
 
 
+
 def setup_ui():
     with gr.Blocks() as demo:
         with gr.Column():