Update app.py

app.py

@@ -31,7 +31,6 @@ def call_gpt4o_mini(api_key, context, user_prompt):
         return "Please provide a valid API key (should start with 'sk-')"
 
     try:
-        # Replace with your actual API endpoint
         url = os.getenv('GPT4O_MINI_API_URL', 'https://api.openai.com/v1/chat/completions')
         headers = {
             "Authorization": f"Bearer {api_key}",
@@ -44,7 +43,7 @@ def call_gpt4o_mini(api_key, context, user_prompt):
         ]
         
         payload = {
-            "model": "gpt-4",  # or your specific model
+            "model": "gpt-4",
             "messages": messages,
             "max_tokens": 500,
             "temperature": 0.7
@@ -148,20 +147,12 @@ def extended_analysis(df):
     except Exception as e:
         return output_paths, f"Error during analysis: {str(e)}"
 
-class ChatHistory(gr.components.Component):
+class ChatHistory:
+    """Simple chat history manager without Gradio component inheritance"""
     def __init__(self):
-        super().__init__()
         self.messages = []
         self.data_summary = ""
-        self._current_df = None
-    
-    @property
-    def current_df(self):
-        return self._current_df
-    
-    @current_df.setter
-    def current_df(self, df):
-        self._current_df = df
+        self.current_df = None
     
     def add_message(self, role, content):
         self.messages.append({"role": role, "content": content})
@@ -180,19 +171,7 @@ class ChatHistory(gr.components.Component):
     def clear(self):
         self.messages = []
         self.data_summary = ""
-        self._current_df = None
-
-    def preprocess(self, payload):
-        """
-        Preprocess the input data before it is passed to the component.
-        """
-        return payload
-
-    def postprocess(self, value):
-        """
-        Postprocess the output data before it is returned from the component.
-        """
-        return value
+        self.current_df = None
 
 def analyze_and_visualize(file, message, history, api_key, chat_history):
     """Main function for data analysis with improved state management"""
@@ -269,38 +248,41 @@ def create_demo():
         4. Type keywords like 'visualize', 'plot', 'analyze' to generate charts
         """)
 
-        api_key = gr.Textbox(
-            label="API Key (Optional)",
-            placeholder="Enter your API key here",
-            type="password"
-        )
-        file_input = gr.File(label="Upload CSV", file_types=[".csv"])
-        chatbot = gr.Chatbot(height=500, type='messages')
-        msg = gr.Textbox(
-            label="Ask about your data",
-            placeholder="e.g., 'Show me the distribution of scores' or 'What insights can you find?'"
-        )
-
-        send_btn = gr.Button("Send")
-        clear_btn = gr.Button("Clear Chat")
-
-        def clear_outputs():
-            chat_history.clear()
-            return [], chat_history
+        with gr.Row():
+            with gr.Column():
+                api_key = gr.Textbox(
+                    label="API Key (Optional)",
+                    placeholder="Enter your API key here",
+                    type="password"
+                )
+                file_input = gr.File(label="Upload CSV", file_types=[".csv"])
+                msg = gr.Textbox(
+                    label="Ask about your data",
+                    placeholder="e.g., 'Show me the distribution of scores' or 'What insights can you find?'"
+                )
+                with gr.Row():
+                    send_btn = gr.Button("Send")
+                    clear_btn = gr.Button("Clear Chat")
+            
+            with gr.Column():
+                chatbot = gr.Chatbot(height=600)
 
         msg.submit(
-            fn=lambda f, m, h, k, ch: analyze_and_visualize(f, m, h or [], k, ch),
+            fn=analyze_and_visualize,
             inputs=[file_input, msg, chatbot, api_key, chat_history],
             outputs=[chatbot, chat_history]
         ).then(lambda: "", None, [msg])
 
         send_btn.click(
-            fn=lambda f, m, h, k, ch: analyze_and_visualize(f, m, h or [], k, ch),
+            fn=analyze_and_visualize,
             inputs=[file_input, msg, chatbot, api_key, chat_history],
             outputs=[chatbot, chat_history]
         ).then(lambda: "", None, [msg])
 
-        clear_btn.click(fn=clear_outputs, inputs=[], outputs=[chatbot, chat_history])
+        clear_btn.click(
+            fn=lambda: ([], ChatHistory()),
+            outputs=[chatbot, chat_history]
+        )
 
         demo.queue()
         return demo
@@ -308,6 +290,6 @@ def create_demo():
 if __name__ == "__main__":
     try:
         demo = create_demo()
-        demo.launch(share=True)
+        demo.launch()
     finally:
         cleanup_temp_files()