Update app.py

app.py

@@ -167,6 +167,29 @@ class BasicAgent:
         except Exception as e:
             print(f"Error processing image: {str(e)}")
             return f"I encountered an error analyzing the image: {str(e)}"
+
+    def _calculate(self, expression):
+        try:
+            # Basic calculator using Python's eval with safety restrictions
+            import math, re
+            safe_dict = {k: v for k, v in math.__dict__.items() if not k.startswith('__')}
+            # Only allow safe math operations
+            if re.match(r'^[\d\s\+\-\*\/\(\)\.\,\^\%\w]+$', expression):
+                result = eval(expression, {"__builtins__": {}}, safe_dict)
+                return f"Calculation result: {result}"
+            return "Invalid calculation expression"
+        except Exception as e:
+            return f"Calculation error: {str(e)}"
+
+    def _read_csv(self, file_url):
+        try:
+            response = requests.get(file_url)
+            import io, csv, pandas as pd
+            df = pd.read_csv(io.StringIO(response.text))
+            summary = f"CSV contains {len(df)} rows, {len(df.columns)} columns.\nColumns: {', '.join(df.columns)}\nSample data: {df.head(3).to_string()}"
+            return summary
+        except Exception as e:
+            return f"CSV reading error: {str(e)}"
     
     def process_question(self, question: str) -> str:
         try:
@@ -175,6 +198,16 @@ class BasicAgent:
             if image_url_match and self.gemini_model:
                 image_url = image_url_match.group(0)
                 return self._process_image_query(question, image_url)
+
+            # Read csv
+            if "calculate" in question.lower() and any(c in question for c in "+-*/"):
+                calculation = re.search(r'calculate\s+([\d\s\+\-\*\/\(\)\.\,\^\%]+)', question, re.IGNORECASE)
+                if calculation:
+                    return self._calculate(calculation.group(1))
+                        
+            csv_url_match = re.search(r'https?://\S+\.csv', question, re.IGNORECASE)
+            if csv_url_match:
+                return self._read_csv(csv_url_match.group(0))
                 
             # Check if this is a request about a YouTube video
             youtube_patterns = ["youtube.com", "youtu.be", "watch youtube", "youtube video"]