Update app.py

app.py

@@ -14,7 +14,8 @@ class DataAnalyzer:
     def __init__(self):
         self.data: Optional[pd.DataFrame] = None
         
-    def create_plot(self, plot_type: str, **kwargs) -> go.Figure:
+    def create_plot(self, plot_type: str, **kwargs) -> str:
+        """Create plot and return HTML string"""
         if self.data is None:
             raise ValueError("No data loaded")
             
@@ -55,8 +56,9 @@ class DataAnalyzer:
             )
         else:
             raise ValueError(f"Unknown plot type: {plot_type}")
-            
-        return fig
+        
+        # Convert plot to HTML string
+        return fig.to_html(include_plotlyjs=True, full_html=False)
 
 class ChatAnalyzer:
     def __init__(self):
@@ -83,12 +85,12 @@ class ChatAnalyzer:
             self.history = [("System", f"Error loading file: {str(e)}")]
             return self.history
     
-    def chat(self, message: str, api_key: str) -> Tuple[List[Tuple[str, str]], List[gr.Plot]]:
+    def chat(self, message: str, api_key: str) -> Tuple[List[Tuple[str, str]], str]:
         if self.analyzer.data is None:
-            return [(message, "Please upload a data file first.")], []
+            return [(message, "Please upload a data file first.")], ""
             
         if not api_key:
-            return [(message, "Please provide an OpenAI API key.")], []
+            return [(message, "Please provide an OpenAI API key.")], ""
             
         try:
             os.environ["OPENAI_API_KEY"] = api_key
@@ -109,7 +111,7 @@ class ChatAnalyzer:
             analysis = completion_response.choices[0].message.content
             
             # Create visualizations
-            figures = []
+            plot_html = ""
             try:
                 # Execute any visualization commands in the analysis
                 exec_globals = {
@@ -117,7 +119,7 @@ class ChatAnalyzer:
                     'df': self.analyzer.data,
                     'px': px,
                     'go': go,
-                    'print': lambda x: figures.append(x) if isinstance(x, go.Figure) else None
+                    'print': lambda x: x  # Will be used to capture plot output
                 }
                 
                 # Extract code blocks
@@ -125,7 +127,9 @@ class ChatAnalyzer:
                 code_blocks = re.findall(r'```python\n(.*?)```', analysis, re.DOTALL)
                 
                 for code in code_blocks:
-                    exec(code, exec_globals)
+                    result = exec(code, exec_globals)
+                    if isinstance(result, (go.Figure, px.Figure)):
+                        plot_html += result.to_html(include_plotlyjs=True, full_html=False)
                     
             except Exception as e:
                 analysis += f"\n\nError creating visualization: {str(e)}"
@@ -133,14 +137,11 @@ class ChatAnalyzer:
             # Update chat history
             self.history.append((message, analysis))
             
-            # Convert figures to Gradio plots
-            plots = [gr.Plot(fig) for fig in figures]
-            
-            return self.history, plots
+            return self.history, plot_html
             
         except Exception as e:
             self.history.append((message, f"Error: {str(e)}"))
-            return self.history, []
+            return self.history, ""
     
     def _get_data_context(self) -> str:
         df = self.analyzer.data
@@ -161,11 +162,11 @@ Example commands:
 ```python
 # Create scatter plot
 fig = px.scatter(df, x='column1', y='column2', title='Analysis')
-print(fig)
+fig  # Return the figure object
 
 # Create histogram
 fig = px.histogram(df, x='column', title='Distribution')
-print(fig)
+fig  # Return the figure object
 ```
 
 Provide analysis and insights along with visualizations."""
@@ -173,7 +174,7 @@ Provide analysis and insights along with visualizations."""
 def create_interface():
     analyzer = ChatAnalyzer()
     
-    with gr.Blocks() as demo:
+    with gr.Blocks(css="footer {visibility: hidden}") as demo:
         gr.Markdown("""
         # Interactive Data Analysis Chat
         Upload your data and chat with AI to analyze it!
@@ -196,8 +197,8 @@ def create_interface():
                 )
                 send = gr.Button("Send")
         
-        # Plot output area
-        plot_output = gr.Plot(label="Visualization")
+        # Plot output area (using HTML for interactive plots)
+        plot_output = gr.HTML(label="Visualization")
         
         # Event handlers
         file.change(