Update app.py

app.py

@@ -63,7 +63,7 @@ def predict(message):
     # Utilizar el nuevo index Doc_1 para generar la respuesta
     respuesta = engine.query(user_message)
     
-    # Respuesta de OpenAI basada en la consulta y agregarla al historial
+    # Inicializa la respuesta de OpenAI
     response = openai.ChatCompletion.create(
         model='gpt-3.5-turbo-16k',  # Use the new model
         max_tokens=1000,
@@ -71,12 +71,16 @@ def predict(message):
         temperature=0.5,
         stream=True
     )
-
+    
     partial_message = ""
     for chunk in respuesta:
         if len(chunk['choices'][0]['delta']) != 0:
             partial_message = partial_message + chunk['choices'][0]['delta']['content']
-            yield partial_message
+    
+    # Concatenar la respuesta de OpenAI con la respuesta del index
+    full_response = partial_message + '\n\n' + response.choices[0].message['content']
+    
+    return full_response
 
 # Define la interfaz de Gradio para el nuevo index Doc_1
 gr.Interface(