Update main.py

main.py

@@ -576,6 +576,47 @@ async def rate_limit_exceeded_handler(request: Request, exc: RateLimitExceeded):
         status_code=429,
         content={"detail": "Too many requests. Please try again later."}
     )
+import gradio as gr
+
+# Gradio interface for visualization
+def gradio_visualize(file, prompt, style="seaborn-v0_8"):
+    # Call your existing FastAPI endpoint
+    with open(file.name, "rb") as f:
+        response = client.post(
+            "/visualize/natural",
+            files={"file": f},
+            data={"prompt": prompt, "style": style}
+        )
+    result = response.json()
+    
+    # Return both image and code
+    return (
+        result["image"],  # Base64 image
+        f"```python\n{result['code']}\n```"  # Code with Markdown formatting
+    )
+
+# Create Gradio interface
+visualization_interface = gr.Interface(
+    fn=gradio_visualize,
+    inputs=[
+        gr.File(label="Upload Excel File", type="filepath"),
+        gr.Textbox(label="Visualization Prompt", placeholder="e.g., 'Show sales by region'"),
+        gr.Dropdown(label="Style", choices=plt.style.available, value="seaborn-v0_8")
+    ],
+    outputs=[
+        gr.Image(label="Generated Visualization"),  # Auto-handles base64
+        gr.Markdown(label="Generated Code")  # Renders code with syntax highlighting
+    ],
+    title="📊 Data Visualizer",
+    description="Upload an Excel file and describe the visualization you want"
+)
+
+# Mount Gradio to your FastAPI app
+app = gr.mount_gradio_app(app, visualization_interface, path="/gradio")
+
+
+
+
 
 # ===== ADD THIS AT THE BOTTOM OF main.py =====
 if __name__ == "__main__":