Update app.py

app.py

@@ -16,9 +16,10 @@ genai.configure(api_key=GOOGLE_API_KEY)
 def transform_history(history):
     """Transforma el historial en el formato esperado por Gemini."""
     new_history = []
-    for chat in history:
-        new_history.append({"role": "user", "content": chat[0]})
-        new_history.append({"role": "assistant", "content": chat[1]})
+    for user, assistant in history:
+        new_history.append({"role": "user", "content": user})
+        if assistant:
+            new_history.append({"role": "assistant", "content": assistant})
     return new_history
 
 def bot(files: Optional[List[str]], model_choice: str, system_instruction: Optional[str], history):
@@ -36,20 +37,27 @@ def bot(files: Optional[List[str]], model_choice: str, system_instruction: Optio
         top_p=0.9
     )
     
+    # Si se incluyen archivos, procesa esa entrada multimodal
+    if files:
+        for file_path in files:
+            with open(file_path, "r") as file:
+                file_content = file.read()
+                chat_history.append({"role": "user", "content": f"Archivo cargado: {file_content}"})
+    
     response = genai.ChatCompletion.create(
         model=model_choice,
-        messages=chat_history + [{"role": "user", "content": history[-1][0]}],
+        messages=chat_history,
         generation_config=generation_config
     )
 
     reply = response['candidates'][0]['content']
     for i in range(len(reply)):
         time.sleep(0.05)
-        yield history + [{"role": "assistant", "content": reply[:i + 1]}]
+        yield history + [[None, reply[:i + 1]]]  # Agrega la respuesta progresivamente al historial
 
 # Interfaz con Gradio
 with gr.Blocks() as demo:
-    chatbot = gr.Chatbot(elem_id="chatbot", bubble_full_width=False, type="messages")
+    chatbot = gr.Chatbot(elem_id="chatbot", bubble_full_width=False, type="interactive")
     
     chat_input = gr.Textbox(
         placeholder="Escribe un mensaje...",
@@ -59,11 +67,13 @@ with gr.Blocks() as demo:
     submit_btn = gr.Button("Enviar")
     system_input = gr.Textbox(placeholder="Instrucción del sistema (opcional)", show_label=True, lines=2)
     model_choice = gr.Dropdown(choices=["gemini-1.5-flash"], value="gemini-1.5-flash", label="Modelo")
+    file_input = gr.File(label="Subir archivo (opcional)", file_types=[".txt", ".md", ".json"])
     
+    # Manejar el envío del mensaje
     submit_btn.click(
-        bot,  # Función que manejará la acción del botón
-        inputs=[chat_input, model_choice, system_input, chatbot],  # Asegúrate de que las entradas sean válidas
-        outputs=chatbot  # El chatbot será actualizado con la respuesta
+        bot,
+        inputs=[file_input, model_choice, system_input, chatbot],
+        outputs=chatbot
     )
     
 demo.launch()