Update app.py

app.py

@@ -1,10 +1,9 @@
-import base64
 import os
 import gradio as gr
-import requests
-import markdownify
 from google import genai
 from google.genai import types
+import requests
+import markdownify
 from urllib.robotparser import RobotFileParser
 from urllib.parse import urlparse
 
@@ -31,9 +30,10 @@ def load_page(url: str) -> str:
     except Exception as e:
         return f"Error loading page: {str(e)}"
 
-# Configure Gemini client
+# Initialize Gemini client
 client = genai.Client(api_key=os.environ.get("GEMINI_API_KEY"))
 MODEL = "gemini-2.0-flash"
+
 TOOLS = [
     types.Tool(
         function_declarations=[
@@ -54,19 +54,14 @@ TOOLS = [
     types.Tool(code_execution=types.ToolCodeExecution())
 ]
 
-SYSTEM_INSTRUCTION = """You are an AI assistant with multiple capabilities:
-1. Web browsing through search and direct page access
-2. Code execution for calculations, simulations, and data analysis
-3. File I/O operations for data processing
-
-Use this decision tree:
-- For factual questions: Use search
-- For time-sensitive data: Use browser tool
-- For math/data processing: Generate and execute code
-- Always explain your reasoning"""
+SYSTEM_INSTRUCTION = """You are an AI assistant with:
+1. Web browsing capabilities
+2. Code execution for calculations
+3. Data analysis skills
+Use the most appropriate tool for each query."""
 
-def format_code_response(parts):
-    """Format code execution parts for Markdown display"""
+def format_response(parts):
+    """Format response parts with proper Markdown formatting"""
     formatted = []
     for part in parts:
         if part.text:
@@ -74,9 +69,7 @@ def format_code_response(parts):
         if part.executable_code:
             formatted.append(f"```python\n{part.executable_code.code}\n```")
         if part.code_execution_result:
-            formatted.append(f"**Result**:\n{part.code_execution_result.output}")
-        if part.inline_data:
-            formatted.append(f"![Generated Image](data:image/png;base64,{base64.b64encode(part.inline_data.data).decode()})")
+            formatted.append(f"**Result**:\n```\n{part.code_execution_result.output}\n```")
     return "\n\n".join(formatted)
 
 def generate_response(user_input):
@@ -97,7 +90,7 @@ def generate_response(user_input):
     response_parts = []
     for part in response.candidates[0].content.parts:
         response_parts.append(part)
-        full_response = format_code_response(response_parts)
+        full_response = format_response(response_parts)
         yield full_response
         
         # Handle function calls
@@ -122,24 +115,35 @@ def generate_response(user_input):
                 final_response = chat.send_message("")
                 for final_part in final_response.candidates[0].content.parts:
                     response_parts.append(final_part)
-                    full_response = format_code_response(response_parts)
+                    full_response = format_response(response_parts)
                     yield full_response
 
-# Gradio Interface
-with gr.Blocks(title="Gemini 2.0 AI Assistant") as demo:
-    gr.Markdown("# 🚀 Gemini 2.0 AI Assistant")
-    gr.Markdown("Web Access • Code Execution • Data Analysis")
+# Create Gradio interface
+with gr.Blocks(
+    title="Gemini AI Assistant",
+    css=""".markdown-output {
+        padding: 20px;
+        border-radius: 5px;
+        background: #f9f9f9;
+    }
+    .markdown-output code {
+        background: #f3f3f3;
+        padding: 2px 5px;
+        border-radius: 3px;
+    }"""
+) as demo:
+    gr.Markdown("# 🚀 Gemini AI Assistant")
+    gr.Markdown("Web • Code • Data Analysis")
     
     with gr.Row():
         input_box = gr.Textbox(
             label="Your Query",
-            placeholder="Ask anything or request code execution...",
+            placeholder="Ask anything...",
             lines=3,
-            max_lines=10,
-            autofocus=True
+            max_lines=10
         )
         output_box = gr.Markdown(
-            label="Assistant Response",
+            label="Response",
             elem_classes="markdown-output"
         )
     
@@ -164,19 +168,4 @@ with gr.Blocks(title="Gemini 2.0 AI Assistant") as demo:
     )
 
 if __name__ == "__main__":
-    demo.launch(
-        server_name="0.0.0.0",
-        server_port=7860,
-        css="""
-        .markdown-output {
-            padding: 20px;
-            border-radius: 5px;
-            background: #f9f9f9;
-        }
-        .markdown-output code {
-            background: #f3f3f3;
-            padding: 2px 5px;
-            border-radius: 3px;
-        }
-        """
-    )
+    demo.launch(server_name="0.0.0.0", server_port=7860)