Update app.py

app.py

@@ -6,27 +6,28 @@ import google.generativeai as genai
 from PIL import Image
 
 def process_file(api_key, file, instructions):
-    # ... [keep the existing process_file function identical] ...
+    # Configure Gemini API
+    genai.configure(api_key=api_key)
+    model = genai.GenerativeModel('gemini-2.5-pro-latest')
 
-# Fixed Gradio interface with compatible theme settings
-with gr.Blocks(theme=gr.themes.Default(spacing_size="lg")) as demo:  # Changed from "xl" to "lg"
-    gr.Markdown("# **HD Data Visualizer**  📊✨")
-    
-    with gr.Row():
-        api_key = gr.Textbox(label="🔑 Gemini API Key", type="password")
-        file = gr.File(label="📁 Upload Dataset", file_types=[".csv", ".xlsx"])
+    # File handling with error prevention
+    try:
+        if file.name.endswith('.csv'):
+            df = pd.read_csv(file.name)
+        else:
+            df = pd.read_excel(file.name)
+    except Exception as e:
+        return [f"File Error: {str(e)}"] * 3
+
+    # Enhanced prompt template
+    prompt = f"""Generate exactly 3 distinct matplotlib visualizations for:
+    Columns: {list(df.columns)}
+    Data types: {dict(df.dtypes)}
+    Sample data: {df.head(3).to_dict()}
     
-    instructions = gr.Textbox(label="💡 Custom Instructions (optional)", 
-                            placeholder="E.g.: Focus on time series patterns...")
-    submit = gr.Button("🚀 Generate Visualizations", variant="primary")
+    Requirements:
+    1. 1920x1080 resolution (figsize=(16,9), dpi=120)
+    2. Professional styling (seaborn, grid, proper labels)
+    3. Diverse chart types (include at least 1 advanced visualization)
     
-    with gr.Row():
-        outputs = [gr.Image(label=f"Visualization {i+1}", width=600) for i in range(3)]
-
-    submit.click(
-        process_file,
-        inputs=[api_key, file, instructions],
-        outputs=outputs
-    )
-
-demo.launch()
+    User instructions: {instructions or 'None provided