Update app.py

app.py

@@ -62,16 +62,21 @@ def wait_for_files_active(files):
 def upload(files: Optional[List[str]], chatbot: CHAT_HISTORY) -> CHAT_HISTORY:
     gemini_files = []
     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)
-        gemini_file = upload_to_gemini(image_path, mime_type="image/jpeg")
+        # Verificar si es una imagen o un archivo PDF
+        if file.endswith('.pdf'):
+            gemini_file = upload_to_gemini(file, mime_type="application/pdf")
+        else:
+            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)
+            gemini_file = upload_to_gemini(image_path, mime_type="image/jpeg")
+        
         gemini_files.append(gemini_file)
 
-    # Wait for files to be ready in Gemini
+    # Esperar a que los archivos se procesen en Gemini
     wait_for_files_active(gemini_files)
     chatbot.append(((gemini_file.uri,), None))
     return chatbot
@@ -99,7 +104,7 @@ def bot(
     )
 
     text_prompt = [chatbot[-1][0]] if chatbot and chatbot[-1][0] and isinstance(chatbot[-1][0], str) else []
-    image_prompt = [preprocess_image(Image.open(file).convert('RGB')) for file in files] if files else []
+    image_prompt = [preprocess_image(Image.open(file).convert('RGB')) for file in files if file.endswith(('.jpg', '.jpeg', '.png'))] if files else []
     model = genai.GenerativeModel(model_choice)
     response = model.generate_content(text_prompt + image_prompt, stream=True, generation_config=generation_config)
 
@@ -121,7 +126,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 Images or PDF", file_count="multiple", file_types=["image", "pdf"], scale=1
 )
 run_button_component = gr.Button(value="Run", variant="primary", scale=1)
 model_choice_component = gr.Dropdown(
@@ -178,7 +183,4 @@ with gr.Blocks() as demo:
         queue=False
     )
 
-demo.queue(max_size=99).launch(debug=False, show_error=True)
-
-
-
+demo.queue(max_size=99).launch(debug=False, show_error=True)