Update app.py

app.py

@@ -45,8 +45,10 @@ 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: ')
             try:
-                chunk_json = json.loads(chunk.decode('utf-8'))
+                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', {})
@@ -59,12 +61,12 @@ def process_response(response, history):
                         token_counter += 1
             except json.JSONDecodeError as e:
                 print("Error decoding JSON response:", e)
-                print("Raw chunk:", chunk)  # Log do chunk bruto para diagnóstico
-                break  # ou trate o erro como achar melhor
+                print("Raw chunk:", chunk_str)  # Log do chunk bruto para diagnóstico
     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():