Update app.py

app.py

@@ -1,7 +1,6 @@
 import os
 import gradio as gr
 import google.generativeai as genai
-import tempfile
 
 # Ensure the API key is set
 api_key = os.getenv("GOOGLE_API_KEY")
@@ -19,17 +18,14 @@ def process_input(text_input, files):
         contents.append({"text": text_input})  # Add text content for Gemini
 
     # Handle files: read and prepare for Gemini API
-    for file in files:
-        # Temporary file storage to read content
-        with tempfile.NamedTemporaryFile(delete=False) as tmp:
-            tmp.write(file.read())
-            tmp.seek(0)  # Rewind file to the beginning for reading
-            
-            # Depending on file type and Gemini requirements, you might adjust how you read the file
-            contents.append({
-                "file": tmp.name,
-                "mime_type": file.type  # MIME type is needed if specific handling is required by Gemini
-            })
+    for file_info in files:
+        file_content = file_info['content']  # Access the file content from the Gradio file dictionary
+        mime_type = file_info['metadata']['mime_type']  # Access MIME type from the metadata
+
+        contents.append({
+            "file": file_content,
+            "mime_type": mime_type  # MIME type is used if specific handling is required by Gemini
+        })
 
     # Call Gemini API to process the collected contents
     try:
@@ -44,7 +40,7 @@ def create_interface():
     with gr.Blocks() as app:
         with gr.Row():
             text_input = gr.Textbox(label="Enter your text:", placeholder="Type your query here...")
-            file_input = gr.File(label="Upload files", file_types=["pdf", "png", "jpg", "mp3", "mp4"], file_count="multiple")
+            file_input = gr.File(label="Upload files", type="file", file_types=["pdf", "png", "jpg", "mp3", "mp4"], count="multiple")
             submit_button = gr.Button("Process")
 
         output = gr.Textbox(placeholder="Response will appear here...")
@@ -57,7 +53,6 @@ def create_interface():
 
     return app
 
-# Run the interface
 if __name__ == "__main__":
     app = create_interface()
-    app.launch()
+    app.launch(debug=True)