Update app.py

app.py

@@ -35,21 +35,22 @@ class DataAnalyzer:
         self.current_iteration = 0
         self.analysis_results = []
         
-    def call_gpt4o_mini(self, prompt: str) -> str:
+    def call_gpt4o_mini(self, prompt: str, system_prompt: str) -> str:
         """Call GPT-4o-mini API with proper error handling"""
         try:
-            headers = {
-                "Authorization": f"Bearer {self.api_key}",
-                "Content-Type": "application/json"
-            }
-            response = requests.post(
-                "https://api.gpt4o-mini.example.com/v1/chat",  # Replace with actual endpoint
-                json={"prompt": prompt, "max_tokens": 500, "temperature": 0.7},
-                headers=headers,
-                timeout=15
+            client = openai.OpenAI(api_key=self.api_key)
+            messages = [
+                {"role": "system", "content": system_prompt},
+                {"role": "user", "content": prompt}
+            ]
+            
+            response = client.chat.completions.create(
+                model="gpt-4o-mini",
+                messages=messages,
+                max_tokens=500,
+                temperature=0.7
             )
-            response.raise_for_status()
-            return response.json()["choices"][0]["text"]
+            return response.choices[0].message.content
         except Exception as e:
             return f"API Error: {str(e)}"
 
@@ -165,17 +166,47 @@ class GradioInterface:
         self.analyzer = None
         self.df = None
         
-    def create_interface(self):
-        with gr.Blocks() as demo:
-            gr.Markdown("# Intelligent Data Analysis Agent")
+    DEFAULT_SYSTEM_PROMPT = """You are an expert data scientist and analyst with the following capabilities:
+1. Deep understanding of statistical analysis and machine learning
+2. Ability to identify patterns and insights in data
+3. Experience with data visualization and interpretation
+4. Strong technical communication skills to explain complex findings
+
+Your task is to:
+1. Analyze data thoroughly and systematically
+2. Identify key patterns and relationships
+3. Suggest appropriate visualizations
+4. Provide clear interpretations of findings
+5. Recommend next steps for deeper analysis
+
+Always explain your reasoning and provide context for your findings."""
+
+def create_interface(self):
+        with gr.Blocks(theme=gr.themes.Soft()) as demo:
+            gr.Markdown("# 🔍 Intelligent Data Analysis Agent")
             
             with gr.Row():
-                api_key = gr.Textbox(label="GPT-4o-mini API Key", type="password")
-                file_input = gr.File(label="Upload CSV file")
-                
+                with gr.Column(scale=1):
+                    api_key = gr.Textbox(
+                        label="GPT-4o-mini API Key",
+                        type="password",
+                        placeholder="sk-..."
+                    )
+                    file_input = gr.File(
+                        label="Upload CSV file"
+                    )
+                    
+                    with gr.Accordion("⚙️ Advanced Settings", open=False):
+                        system_prompt = gr.TextArea(
+                            label="System Prompt",
+                            value=DEFAULT_SYSTEM_PROMPT,
+                            lines=8
+                        )
+                        
             with gr.Row():
-                analysis_notes = gr.Textbox(label="Analysis Notes (Optional)", 
-                                          placeholder="Any specific analysis preferences...")
+                analysis_notes = gr.Textbox(
+                    label="Analysis Notes (Optional)", 
+                    placeholder="Any specific analysis preferences...")
                 
             with gr.Row():
                 analyze_btn = gr.Button("Analyze Data")
@@ -184,7 +215,7 @@ class GradioInterface:
             output_text = gr.Markdown()
             output_gallery = gr.Gallery()
             
-            def analyze(api_key, file, notes):
+            def analyze(api_key, file, notes, system_prompt):
                 if not api_key or not file:
                     return "Please provide both API key and data file.", None