Update app.py

app.py

@@ -102,10 +102,48 @@ class CSVChatApp:
         else:
             return "No CSV files were uploaded."
     
+    # def process_query(self, query, history):
+    #     """Process a user query and generate a response."""
+    #     if not self.uploaded_files:
+    #         return "Please upload CSV files before asking questions."
+        
+    #     # Add user message to history
+    #     self.chat_history.append({"role": "user", "content": query})
+        
+    #     # Process the query
+    #     try:
+    #         response = self.query_engine.query(query)
+    #         answer = response["answer"]
+            
+    #         # Check if response contains an image
+    #         if isinstance(answer, dict) and "image" in answer:
+    #             # Handle image in response
+    #             img_data = answer["image"]
+    #             img = Image.open(io.BytesIO(base64.b64decode(img_data)))
+    #             img_path = EXPORT_DIR / f"viz_{int(time.time())}.png"
+    #             img.save(img_path)
+                
+    #             # Update answer to include image path
+    #             text_response = answer.get("text", "Generated visualization")
+    #             answer = (text_response, str(img_path))
+            
+    #         # Add assistant message to history
+    #         self.chat_history.append({"role": "assistant", "content": answer})
+            
+    #         return answer
+            
+    #     except Exception as e:
+    #         error_msg = f"Error processing query: {str(e)}"
+    #         self.chat_history.append({"role": "assistant", "content": error_msg})
+    #         return error_msg
+
+    # In app.py, modify the process_query method:
+
     def process_query(self, query, history):
         """Process a user query and generate a response."""
         if not self.uploaded_files:
-            return "Please upload CSV files before asking questions."
+            # Return in the correct format for the chatbot
+            return history + [[query, "Please upload CSV files before asking questions."]]
         
         # Add user message to history
         self.chat_history.append({"role": "user", "content": query})
@@ -130,12 +168,16 @@ class CSVChatApp:
             # Add assistant message to history
             self.chat_history.append({"role": "assistant", "content": answer})
             
-            return answer
+            # Return in the correct format for the chatbot
+            return history + [[query, answer]]
             
         except Exception as e:
             error_msg = f"Error processing query: {str(e)}"
             self.chat_history.append({"role": "assistant", "content": error_msg})
-            return error_msg
+            
+            # Return in the correct format for the chatbot
+            return history + [[query, error_msg]]
+
     
     def export_conversation(self):
         """Export the conversation as a report."""