Update app.py

app.py

@@ -4,8 +4,8 @@ import pandas as pd
 import numpy as np
 import requests
 import json
-import time # Ensure time is imported for backoff
-import matplotlib.pyplot as plt # Explicitly import matplotlib.pyplot here for use in exec() scope
+import time 
+import matplotlib.pyplot as plt # Explicitly imported for use in exec() scope
 
 # --- CONFIG ---
 # Note: GEMINI_API_KEY is retrieved from environment variables/secrets.
@@ -17,7 +17,8 @@ if not GEMINI_API_KEY:
 
 # Define API endpoints and models
 GEMINI_BASE = "https://generativelanguage.googleapis.com/v1beta"
-CHAT_MODEL = "gemini-2.5-flash-preview-09-2025" # Using the correct model for structured output
+# Using the correct model for structured output
+CHAT_MODEL = "gemini-2.5-flash-preview-09-2025" 
 EMBED_MODEL = "models/embedding-001" 
 
 # Define the JSON schema for structured output
@@ -52,7 +53,8 @@ SYSTEM_INSTRUCTION = (
 def chat_with_gemini(prompt, context):
     """Sends a prompt and data context to the Gemini model for structured analysis (reasoning + code)."""
     
-    url = f"{GEMINI_BASE}/{CHAT_MODEL}:generateContent?key={GEMINI_API_KEY}"
+    # FIX: Correctly prepend 'models/' to the model name in the URL path
+    url = f"{GEMINI_BASE}/models/{CHAT_MODEL}:generateContent?key={GEMINI_API_KEY}"
     
     # Construct the full prompt including the data context
     full_prompt = f"Data Context (DataFrame Head and Columns):\n{context}\n\nUser Question: {prompt}"
@@ -73,7 +75,7 @@ def chat_with_gemini(prompt, context):
     for attempt in range(max_retries):
         try:
             r = requests.post(url, headers={'Content-Type': 'application/json'}, data=json.dumps(payload))
-            r.raise_for_status()
+            r.raise_for_status() # This will raise an exception for 4xx or 5xx errors
             data = r.json()
             
             # The JSON output is a string inside the 'text' part