Update app.py

app.py

@@ -48,7 +48,6 @@ engine = index.as_query_engine(query_engine_tools=query_engine_tools, service_co
 # Prompt personalizado para el nuevo index
 prompt = 'responder en español como un asistente experto en medicina, dando una respuesta detallada y reflejando de forma fiel los datos disponibles. Considerar segun aplique la siguiente informacion adicional: Nombre del Medicamento: Acsodix. Principio Activo: Vortioxetina. Formas y Dosis: Acsodix 5, 10, y 20 mg (comprimidos recubiertos). Indicaciones: Tratamiento de episodios de depresión mayor en adultos. Contraindicaciones: Alergia a la Vortioxetina, toma de ciertos inhibidores de la monoaminooxidasa, enfermedad hepática grave, etc. Precauciones: Embarazo, lactancia, conducción, uso de otros medicamentos, etc. Efectos Adversos: Náuseas, diarrea, vómitos, mareo, prurito, etc. Instrucciones de Uso: Se puede tomar con o sin alimentos, no se aconseja la combinación con alcohol, etc. Sobredosis: Síntomas y tratamientos en caso de sobredosis. Presentación: Cajas conteniendo 30 comprimidos recubiertos en diferentes dosis. Almacenamiento: Conservar a temperatura ambiente inferior a 30 °C. Fabricante y Distribuidor: Mega Labs S.A., Industria Uruguaya, La dosis máxima de Acsodix para adultos menores a 65 años es de 20 mg al día. Se debe tener precaución cuando se traten pacientes mayores de 65 años de edad con dosis superiores a 10 mg de Vortioxetina una vez al día, ya que los datos son limitados.'
 
-# Funcion de chat para el nuevo index Doc_1
 def predict(message):
     # Agrega la pregunta del usuario al historial en formato OpenAI
     history_openai_format = [{"role": "user", "content": message}]
@@ -74,7 +73,12 @@ def predict(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 the complete response as a list (Gradio expects a list)
+    return [full_response]
 
 # Define la interfaz de Gradio para el nuevo index Doc_1
 gr.Interface(