Update app.py

app.py

@@ -10,6 +10,7 @@ import google.generativeai as genai
 import gradio as gr
 from PIL import Image
 from dotenv import load_dotenv
+from PyPDF2 import PdfReader
 
 # Cargar las variables de entorno desde el archivo .env
 load_dotenv()
@@ -39,15 +40,38 @@ def cache_pil_image(image: Image.Image) -> str:
     image.save(image_path, "JPEG")
     return image_path
 
+def process_text_file(file_path: str):
+    with open(file_path, 'r') as file:
+        return file.read()
+
+def process_pdf_file(file_path: str):
+    reader = PdfReader(file_path)
+    text = ""
+    for page in reader.pages:
+        text += page.extract_text()
+    return text
+
 def upload(files: Optional[List[str]], chatbot: CHAT_HISTORY) -> CHAT_HISTORY:
     for file in files:
-        image = Image.open(file).convert('RGB')
-        image_preview = preprocess_image(image)
-        if image_preview:
-            # Display a preview of the uploaded image
-            gr.Image(image_preview).render()
-        image_path = cache_pil_image(image)
-        chatbot.append(((image_path,), None))
+        file_extension = file.split('.')[-1].lower()
+        
+        if file_extension in ['jpg', 'jpeg', 'png', 'gif', 'bmp']:  # Imágenes
+            image = Image.open(file).convert('RGB')
+            image_preview = preprocess_image(image)
+            if image_preview:
+                # Display a preview of the uploaded image
+                gr.Image(image_preview).render()
+            image_path = cache_pil_image(image)
+            chatbot.append(((image_path,), None))
+
+        elif file_extension in ['txt']:  # Archivos de texto
+            text_content = process_text_file(file)
+            chatbot.append(((text_content,), None))
+
+        elif file_extension in ['pdf']:  # Archivos PDF
+            pdf_content = process_pdf_file(file)
+            chatbot.append(((pdf_content,), None))
+        
     return chatbot
 
 def user(text_prompt: str, chatbot: CHAT_HISTORY):
@@ -95,7 +119,7 @@ text_prompt_component = gr.Textbox(
     placeholder="Message...", show_label=False, autofocus=True, scale=8
 )
 upload_button_component = gr.UploadButton(
-    label="Upload Images", file_count="multiple", file_types=["image"], scale=1
+    label="Upload Files", file_count="multiple", file_types=["image", "text", "pdf"], scale=1
 )
 run_button_component = gr.Button(value="Run", variant="primary", scale=1)
 model_choice_component = gr.Dropdown(