Update app.py

app.py

@@ -41,15 +41,21 @@ if __name__ == '__main__':
     )
     logger.info(f"chatpdf model: {model}")
 
-    # Función para generar respuesta (sin yield)
-    def predict(message, history):
+    def predict_stream(message, history):
         history_format = []
         for human, assistant in history:
             history_format.append([human, assistant])
         model.history = history_format
+        for chunk in model.predict_stream(message):
+            yield chunk
+
+
+    def predict(message, history):
+        logger.debug(message)
         response, reference_results = model.predict(message)
         r = response + "\n\n" + '\n'.join(reference_results)
-        return r, history_format
+        logger.debug(r)
+        return r
 
     # Crear el chatbot
     chatbot_stream = gr.Chatbot(
@@ -66,11 +72,12 @@ if __name__ == '__main__':
     examples = ['Puede hablarme del PNL?', 'Introducción a la PNL']
 
     # Crear la interfaz sin utilizar la función que usa yield
-    chat_interface = gr.Interface(
-        fn=predict,
-        inputs=[gr.Textbox(lines=4, placeholder="Ask me question", scale=7), gr.State()],
-        outputs=[chatbot_stream, gr.State()],
+   chat_interface_stream = gr.ChatInterface(
+        predict_stream,
+        textbox=gr.Textbox(lines=4, placeholder="Ask me question", scale=7),  # Añadir submit=True
         title=title,
+        # description=description,
+        chatbot=chatbot_stream,
         css=css,
         examples=examples,
         theme='soft',