Update app.py

app.py

@@ -6,80 +6,87 @@ import google.generativeai as genai
 from PIL import Image
 
 def process_file(api_key, file, instructions):
-    # Configure Gemini API
     genai.configure(api_key=api_key)
-    model = genai.GenerativeModel('gemini-2.5-pro-preview-03-25')
+    model = genai.GenerativeModel('gemini-2.5-pro-latest')
 
-    # Read uploaded file
-    if file.name.endswith('.csv'):
-        df = pd.read_csv(file.name)
-    else:
-        df = pd.read_excel(file.name)
+    # Read file with error handling
+    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
 
-    # Generate analysis prompt
-    data_description = df.describe().to_string()
-    columns_info = "\n".join([f"{col}: {df[col].dtype}" for col in df.columns])
-    prompt = f"""
-    Given this dataset:
-    Columns and types:
-    {columns_info}
+    # Enhanced prompt with visualization requirements
+    prompt = f"""Generate exactly 3 distinct matplotlib visualization codes for this dataset:
+    Columns: {list(df.columns)}
+    Data types: {dict(df.dtypes)}
+    Sample data: {df.head(3).to_dict()}
     
-    Data summary:
-    {data_description}
+    Requirements:
+    1. 1920x1080 resolution (figsize=(16,9), dpi=120)
+    2. Professional styling (seaborn style, grid lines)
+    3. Clear labels and titles
+    4. Diverse chart types (at least 1 must be non-basic)
     
-    User instructions: {instructions if instructions else 'No specific instructions provided.'}
+    User instructions: {instructions or 'No specific instructions'}
     
-    Generate Python code for 3 different matplotlib visualizations. Only provide executable code without explanations.
-    Format as:
+    Format response strictly as:
     # Visualization 1
+    plt.figure(figsize=(16,9), dpi=120)
     [code]
+    plt.tight_layout()
+    
     # Visualization 2
-    [code]
-    # Visualization 3
-    [code]
+    ...
     """
 
-    # Get Gemini response
     response = model.generate_content(prompt)
-    code = response.text.strip()
+    code_blocks = response.text.split("# Visualization ")[1:4]  # Force 3 max
 
-    # Process visualizations
     visualizations = []
-    for i, viz_code in enumerate(code.split("# Visualization")[1:4], 1):
-        plt.figure(figsize=(10, 6))
+    for i, block in enumerate(code_blocks, 1):
         try:
-            # Clean and execute generated code
-            cleaned_code = '\n'.join(line.strip() for line in viz_code.split('\n') if line.strip())
-            exec(cleaned_code, {'df': df, 'plt': plt})
-            plt.title(f"Visualization {i}")
+            # HD configuration
+            plt.figure(figsize=(16, 9), dpi=120)
+            plt.style.use('seaborn')
             
-            # Convert plot to Gradio-compatible format
+            # Clean and execute code
+            cleaned_code = '\n'.join([line.strip() for line in block.split('\n')[1:] if line.strip()])
+            exec(cleaned_code, {'df': df, 'plt': plt})
+            plt.title(f"Visualization {i}", fontsize=14)
+            plt.tight_layout()
+
+            # Save HD image
             buf = io.BytesIO()
-            plt.savefig(buf, format='png', bbox_inches='tight')
+            plt.savefig(buf, format='png', dpi=120, bbox_inches='tight')
             plt.close()
             buf.seek(0)
             visualizations.append(Image.open(buf))
         except Exception as e:
-            print(f"Visualization {i} error: {str(e)}")
-            visualizations.append(None)
+            print(f"Visualization {i} failed: {str(e)}")
+            visualizations.append(Image.new('RGB', (1920, 1080), color=(73, 109, 137)))
+
+    # Ensure exactly 3 outputs with fallback
+    while len(visualizations) < 3:
+        visualizations.append(Image.new('RGB', (1920, 1080), color=(73, 109, 137)))
 
-    # Ensure 3 outputs for Gradio
-    return visualizations + [None]*(3-len(visualizations))
+    return visualizations[:3]
 
-# Gradio interface
-with gr.Blocks() as demo:
-    gr.Markdown("""# AI-Powered Data Visualizer
-    Upload your data file and get automatic visualizations powered by Gemini""")
+# Gradio interface with HD layout
+with gr.Blocks(theme=gr.themes.Default(spacing_size="xl")) as demo:
+    gr.Markdown("# **HD Data Visualizer**  📊✨")
     
     with gr.Row():
-        api_key = gr.Textbox(label="Gemini API Key", type="password")
-        instructions = gr.Textbox(label="Custom Instructions (optional)")
+        api_key = gr.Textbox(label="🔑 Gemini API Key", type="password")
+        file = gr.File(label="📁 Upload Dataset", file_types=[".csv", ".xlsx"])
     
-    file = gr.File(label="Dataset (CSV/Excel)", file_types=[".csv", ".xlsx"])
-    submit = gr.Button("Generate Insights", variant="primary")
+    instructions = gr.Textbox(label="💡 Custom Instructions (optional)", placeholder="E.g.: Focus on time series patterns...")
+    submit = gr.Button("🚀 Generate Visualizations", variant="primary")
     
     with gr.Row():
-        outputs = [gr.Image(label=f"Visualization {i+1}") for i in range(3)]
+        outputs = [gr.Image(label=f"Visualization {i+1}", width=600) for i in range(3)]
 
     submit.click(
         process_file,