Update app.py

app.py

@@ -1,71 +1,55 @@
 import os
-import time
-import mimetypes
-import google.generativeai as genai
 import gradio as gr
+import google.generativeai as genai
 
-# Configure the Gemini API
-genai.configure(api_key=os.getenv("GOOGLE_API_KEY"))
-
-def upload_to_gemini(file_path, mime_type=None):
-    """Uploads the given file to Gemini, returning the URI for use in chat."""
-    file = genai.upload_file(file_path, mime_type=mime_type)
-    print(f"Uploaded file '{file.display_name}' as: {file.uri}")
-    return file.uri
-
-def wait_for_files_active(file_uris):
-    """Ensure all files are processed and ready for use."""
-    print("Waiting for file processing...")
-    for uri in file_uris:
-        while True:
-            file = genai.get_file(uri)
-            if file.state.name == "ACTIVE":
-                print("File ready.")
-                break
-            elif file.state.name == "FAILED":
-                raise Exception(f"File {uri} failed to process")
-            time.sleep(10)
-
-def handle_files(files):
-    """Handle file uploads, process them, and interact with Gemini API."""
-    # Save uploaded files locally
-    saved_files = []
-    for file in files:
-        local_path = f"/tmp/{file.name}"
-        with open(local_path, "wb") as f:
-            f.write(file.read())
-        saved_files.append((local_path, mimetypes.guess_type(local_path)[0]))
-
-    # Upload files to Gemini and wait for them to be ready
-    file_uris = [upload_to_gemini(path, mime_type) for path, mime_type in saved_files]
-    wait_for_files_active(file_uris)
+# Initialize the Gemini API
+def initialize_gemini():
+    api_key = os.getenv("GOOGLE_API_KEY")
+    if not api_key:
+        raise ValueError("GOOGLE_API_KEY is not set in environment variables.")
+    genai.configure(api_key=api_key)
+    return genai.GenerativeModel("gemini-2.0-flash-exp")  # Assuming gemini-2.0-flash-exp is the correct model name
+
+# Handle file inputs and text to generate a response
+def generate_response(text_input, files):
+    contents = []
+    if text_input:
+        contents.append({"text": text_input})  # Add text input to the contents
     
-    # Start a chat session with uploaded files
-    model = genai.GenerativeModel(
-        model_name="gemini-2.0-flash-exp",
-        generation_config={
-            "temperature": 0.4,
-            "top_p": 0.95,
-            "top_k": 40,
-            "max_output_tokens": 8192,
-            "response_mime_type": "text/plain",
-        }
-    )
-    chat_session = model.start_chat(history=[{"role": "user", "parts": [{"file_uri": uri} for uri in file_uris]}])
-    response = chat_session.send_message("Describe the contents.")
+    for file in files:
+        with open(file.name, "rb") as f:
+            data = f.read()
+        # Guess the MIME type from the file input
+        mime_type = gr.io.utils.guess_mime_type(file.name)
+        if mime_type:
+            contents.append({
+                "mime_type": mime_type,
+                "data": data
+            })
+
+    model = initialize_gemini()
+    response = model.generate_content(contents)
+    response.resolve()
     return response.text
 
-def gradio_interface():
-    """Setup Gradio UI for file uploads and processing."""
+# Gradio Interface
+def setup_interface():
     with gr.Blocks() as app:
         with gr.Row():
-            file_input = gr.File(label="Upload files", file_types=[".pdf", ".png", ".jpg", ".mp3", ".mp4"])
-            submit_button = gr.Button("Submit")
-        output = gr.Textbox(label="Response from Gemini")
+            text_input = gr.Textbox(label="Enter any text:", placeholder="Type your question or comment here...")
+            file_input = gr.File(label="Upload your files", file_types=["image", "video", "audio", "pdf"])
+            submit_button = gr.Button("Generate Response")
+        response_output = gr.Textbox(label="Response from Gemini", placeholder="Response will appear here...")
 
-        submit_button.click(fn=handle_files, inputs=[file_input], outputs=[output])
+        submit_button.click(
+            fn=generate_response,
+            inputs=[text_input, file_input],
+            outputs=[response_output]
+        )
 
     return app
 
-app = gradio_interface()
-app.launch()
+# Run the interface
+if __name__ == "__main__":
+    app = setup_interface()
+    app.launch(share=True)  # Set `share=True` for Hugging Face Spaces or public deployment