Update app.py

app.py

@@ -444,12 +444,26 @@ with gr.Blocks(theme=theme) as demo:
             download_button = gr.File(label="Descargar Informe sobre Administración de Medicamentos a Recién Nacidos", value="Reporte.pdf")
             
             # Chatbot
-            chatbot_output = gr.Chatbot(label="ChatBot")
-            chatbot_input = gr.Textbox(label="Tu mensaje")
-            submit_button = gr.Button("Enviar")
-            chatbot_history = gr.State(value=[])
-            image_url = gr.State(value=None)
+            #chatbot_output = gr.Chatbot(label="ChatBot")
+            #chatbot_input = gr.Textbox(label="Tu mensaje")
+            #submit_button = gr.Button("Enviar")
+            #chatbot_history = gr.State(value=[])
+            #image_url = gr.State(value=None)
+
+            with gr.Blocks() as demo:
+                with gr.Row():
+                    with gr.Column(scale=1):
+                        chatbot_output = gr.Chatbot(label="ChatBot", elem_id="chatbot_output")
+                        chatbot_input = gr.Textbox(label="Tu mensaje", elem_id="chatbot_input")
+                        submit_button = gr.Button("Enviar")
+                        chatbot_history = gr.State(value=[])
+                    
+                    with gr.Column(scale=1):
+                        image_url = gr.Image(value=None, label="Image URL", elem_id="image_display")
+            
+                submit_button.click(chatbot_response, [chatbot_input, chatbot_history], [chatbot_output, image_url])
 
+            
             # Main accordion for categories
             with gr.Accordion("Categorías de Preguntas", open=True):