Update app.py

app.py

@@ -39,46 +39,15 @@ def cache_pil_image(image: Image.Image) -> str:
     image.save(image_path, "JPEG")
     return image_path
 
-def upload_to_gemini(path, mime_type=None):
-    """Uploads the given file to Gemini."""
-    file = genai.upload_file(path, mime_type=mime_type)
-    print(f"Uploaded file '{file.display_name}' as: {file.uri}")
-    return file
-
-def wait_for_files_active(files):
-    """Waits for the given files to be active."""
-    print("Waiting for file processing...")
-    for name in (file.name for file in files):
-        file = genai.get_file(name)
-        while file.state.name == "PROCESSING":
-            print(".", end="", flush=True)
-            time.sleep(10)
-            file = genai.get_file(name)
-        if file.state.name != "ACTIVE":
-            raise Exception(f"File {file.name} failed to process")
-    print("...all files ready")
-    print()
-
 def upload(files: Optional[List[str]], chatbot: CHAT_HISTORY) -> CHAT_HISTORY:
-    gemini_files = []
     for file in files:
-        # Verificar si es una imagen o un archivo PDF
-        if file.name.endswith('.pdf'):
-            gemini_file = upload_to_gemini(file.name, mime_type="application/pdf")
-        else:
-            image = Image.open(file.name).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)
-
-    # Esperar a que los archivos se procesen en Gemini
-    wait_for_files_active(gemini_files)
-    chatbot.append(((gemini_file.uri,), None))
+        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))
     return chatbot
 
 def user(text_prompt: str, chatbot: CHAT_HISTORY):
@@ -104,7 +73,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.name).convert('RGB')) for file in files if file.name.endswith(('.jpg', '.jpeg', '.png'))] if files else []
+    image_prompt = [preprocess_image(Image.open(file).convert('RGB')) for file in files] if files else []
     model = genai.GenerativeModel(model_choice)
     response = model.generate_content(text_prompt + image_prompt, stream=True, generation_config=generation_config)
 
@@ -125,8 +94,8 @@ chatbot_component = gr.Chatbot(
 text_prompt_component = gr.Textbox(
     placeholder="Message...", show_label=False, autofocus=True, scale=8
 )
-upload_button_component = gr.File(
-    label="Upload Images or PDF", file_count="multiple", file_types=["image", "pdf"], scale=1
+upload_button_component = gr.UploadButton(
+    label="Upload Images", file_count="multiple", file_types=["image"], scale=1
 )
 run_button_component = gr.Button(value="Run", variant="primary", scale=1)
 model_choice_component = gr.Dropdown(