Update app.py

app.py

@@ -3,6 +3,7 @@ import gradio as gr
 from gradio import ChatMessage
 from typing import Iterator
 import google.generativeai as genai
+import time # Import time module for potential debugging/delay
 
 # get Gemini API Key from the environ variable
 GEMINI_API_KEY = os.getenv("GEMINI_API_KEY")
@@ -26,27 +27,62 @@ def format_chat_history(messages: list) -> list:
             })
     return formatted_history
 
-def stream_gemini_response(user_message: str, messages: list) -> Iterator[list]:
+def stream_gemini_response(message_input: str|gr.File, messages: list) -> Iterator[list]:
     """
-    Streams thoughts and response with conversation history support.
+    Streams thoughts and response with conversation history support, handling text or file input.
     """
-    try:
-        print(f"\n=== New Request ===")
+    user_message = ""
+    input_file = None
+
+    if isinstance(message_input, str):
+        user_message = message_input
+        print(f"\n=== New Request (Text) ===")
         print(f"User message: {user_message}")
+        if not user_message.strip(): # Robust check: if text message is empty or whitespace
+            messages.append(ChatMessage(role="assistant", content="Please provide a non-empty text message or upload a file.")) # More specific message
+            yield messages
+            return
+
+    elif isinstance(message_input, gr.File): #gr.File directly should be used with newer gradio versions (v4+)
+        input_file = message_input.name # Access the temporary file path
+        file_type = message_input.original_name.split('.')[-1].lower() #Get original filename's extension
+        print(f"\n=== New Request (File) ===")
+        print(f"File uploaded: {input_file}, type: {file_type}")
+
+        try:
+            with open(input_file, "rb") as f: #Open file in binary mode for universal handling
+                file_data = f.read()
+
+            if file_type in ['png', 'jpg', 'jpeg', 'gif']: #Example Image Types - expand as needed
+                user_message = {"inline_data": {"mime_type": f"image/{file_type}", "data": file_data}} #Prepare image part for Gemini
+            elif file_type == 'csv':
+                user_message = {"inline_data": {"mime_type": "text/csv", "data": file_data}} #Prepare csv part
+
+        except Exception as e:
+            print(f"Error reading file: {e}")
+            messages.append(ChatMessage(role="assistant", content=f"Error reading file: {e}"))
+            yield messages
+            return
+    else:
+        messages.append(ChatMessage(role="assistant", content="Sorry, I cannot understand this input format. Please use text or upload a valid file.")) # More informative error
+        yield messages
+        return
+
 
+    try:
         # Format chat history for Gemini
         chat_history = format_chat_history(messages)
 
         # Initialize Gemini chat
         chat = model.start_chat(history=chat_history)
-        response = chat.send_message(user_message, stream=True)
+        response = chat.send_message(user_message, stream=True) #Send the message part as is
 
-        # Initialize buffers and flags
+        # Initialize buffers and flags - same as before
         thought_buffer = ""
         response_buffer = ""
         thinking_complete = False
 
-        # Add initial thinking message
+        # Add initial thinking message - same as before
         messages.append(
             ChatMessage(
                 role="assistant",
@@ -55,7 +91,7 @@ def stream_gemini_response(user_message: str, messages: list) -> Iterator[list]:
             )
         )
 
-        for chunk in response:
+        for chunk in response: #streaming logic - same as before
             parts = chunk.candidates[0].content.parts
             current_chunk = parts[0].text
 
@@ -103,6 +139,7 @@ def stream_gemini_response(user_message: str, messages: list) -> Iterator[list]:
                     content=thought_buffer,
                     metadata={"title": "⚙️ Thinking: *The thoughts produced by the model are experimental"}
                 )
+            #time.sleep(0.05) #Optional: Uncomment this line to add a slight delay for debugging/visualization of streaming. Remove for final version
 
             yield messages
 
@@ -118,11 +155,11 @@ def stream_gemini_response(user_message: str, messages: list) -> Iterator[list]:
         )
         yield messages
 
-def user_message(msg: str, history: list) -> tuple[str, list]:
+def user_message(message_text, file_upload, history: list) -> tuple[str, None, list]:
     """Adds user message to chat history"""
-    history.append(ChatMessage(role="user", content=msg))
-    return "", history
-
+    msg = message_text if message_text else file_upload
+    history.append(ChatMessage(role="user", content=msg if isinstance(msg, str) else msg.name)) #Store message or filename in history.
+    return "", None, history #clear both input fields
 
 # Create the Gradio interface
 with gr.Blocks(theme=gr.themes.Soft(primary_hue="teal", secondary_hue="slate", neutral_hue="neutral")) as demo:
@@ -130,7 +167,7 @@ with gr.Blocks(theme=gr.themes.Soft(primary_hue="teal", secondary_hue="slate", n
 
     chatbot = gr.Chatbot(
         type="messages",
-        label="Gemini2.0 'Thinking' Chatbot",
+        label="Gemini2.0 'Thinking' Chatbot (Streaming Output)", #Label now indicates streaming
         render_markdown=True,
         scale=1,
         avatar_images=(None,"https://lh3.googleusercontent.com/oxz0sUBF0iYoN4VvhqWTmux-cxfD1rxuYkuFEfm1SFaseXEsjjE4Je_C_V3UQPuJ87sImQK3HfQ3RXiaRnQetjaZbjJJUkiPL5jFJ1WRl5FKJZYibUA=w214-h214-n-nu")
@@ -141,8 +178,9 @@ with gr.Blocks(theme=gr.themes.Soft(primary_hue="teal", secondary_hue="slate", n
             lines=1,
             label="Chat Message",
             placeholder="Type your message here...",
-            scale=4
+            scale=3
         )
+        file_upload = gr.File(label="Upload File", file_types=["image", ".csv"], scale=2) # Allow image and CSV files
 
         clear_button = gr.Button("Clear Chat", scale=1)
 
@@ -157,7 +195,7 @@ with gr.Blocks(theme=gr.themes.Soft(primary_hue="teal", secondary_hue="slate", n
 
     gr.Examples(
         examples=example_prompts,
-        inputs=input_box,
+        inputs=[input_box],
         label="Examples: Get Gemini to show its thinking process with these prompts!",
         examples_per_page=5 # Adjust as needed
     )
@@ -166,22 +204,30 @@ with gr.Blocks(theme=gr.themes.Soft(primary_hue="teal", secondary_hue="slate", n
     # Set up event handlers
     msg_store = gr.State("")  # Store for preserving user message
 
+
     input_box.submit(
-        lambda msg: (msg, msg, ""),  # Store message and clear input
-        inputs=[input_box],
-        outputs=[msg_store, input_box, input_box],
+        user_message,
+        inputs=[input_box, file_upload, chatbot],
+        outputs=[input_box, file_upload, chatbot],
         queue=False
     ).then(
-        user_message,  # Add user message to chat
-        inputs=[msg_store, chatbot],
-        outputs=[input_box, chatbot],
+        stream_gemini_response,
+        inputs=[input_box, chatbot], # Input either from text box or file, logic inside stream_gemini_response
+        outputs=chatbot
+    )
+
+    file_upload.upload(
+        user_message,
+        inputs=[input_box, file_upload, chatbot], # even textbox is input here so clearing both will work
+        outputs=[input_box, file_upload, chatbot],
         queue=False
     ).then(
-        stream_gemini_response,  # Generate and stream response
-        inputs=[msg_store, chatbot],
+        stream_gemini_response,
+        inputs=[file_upload, chatbot], # Input is now the uploaded file.
         outputs=chatbot
     )
 
+
     clear_button.click(
         lambda: ([], "", ""),
         outputs=[chatbot, input_box, msg_store],
@@ -202,14 +248,15 @@ with gr.Blocks(theme=gr.themes.Soft(primary_hue="teal", secondary_hue="slate", n
         *   Powered by Google's **Gemini 2.0 Flash** model.
         *   Shows the model's **thoughts** before the final answer (experimental feature).
         *   Supports **conversation history** for multi-turn chats.
+        *   Supports **Image and CSV file uploads** for analysis.
         *   Uses **streaming** for a more interactive experience.
         **Instructions:**
-        1.  Type your message in the input box below or select an example.
-        2.  Press Enter or click Submit to send.
+        1.  Type your message in the input box or Upload a file below.
+        2.  Press Enter/Submit or Upload to send.
         3.  Observe the chatbot's "Thinking" process followed by the final response.
         4.  Use the "Clear Chat" button to start a new conversation.
 
-        *Please note*: The 'thinking' feature is experimental and the quality of thoughts may vary.
+        *Please note*: The 'thinking' feature is experimental and the quality of thoughts may vary. File analysis capabilities may be limited depending on the model's experimental features.
         """
     )