Update app.py

app.py

@@ -200,7 +200,37 @@ class BasicAgent:
             return summary
         except Exception as e:
             return f"Excel reading error: {str(e)}"
-        
+
+    def _read_code_file(self, file_url):
+        try:
+            response = requests.get(file_url)
+            code_content = response.text
+            
+            # Determine file extension for language detection
+            file_extension = file_url.split('.')[-1].lower()
+            
+            # Map common extensions to language names
+            language_map = {
+                'py': 'Python', 'js': 'JavaScript', 'html': 'HTML', 'css': 'CSS',
+                'java': 'Java', 'c': 'C', 'cpp': 'C++', 'cs': 'C#',
+                'php': 'PHP', 'rb': 'Ruby', 'go': 'Go', 'rs': 'Rust',
+                'ts': 'TypeScript', 'sh': 'Shell', 'sql': 'SQL'
+            }
+            
+            language = language_map.get(file_extension, 'Unknown')
+            
+            # Count lines of code
+            line_count = code_content.count('\n') + 1
+            
+            # Create a summary of the code file
+            summary = f"Code file ({language}) with {line_count} lines.\n"
+            summary += f"First 10 lines:\n{'\n'.join(code_content.split('\n')[:10])}"
+            
+            return summary
+        except Exception as e:
+            return f"Code file reading error: {str(e)}"
+
+           
     def process_question(self, question: str) -> str:
         try:
             # Processing image
@@ -214,6 +244,13 @@ class BasicAgent:
             if excel_url_match:
                 return self._read_excel(excel_url_match.group(0))
 
+            # Code
+            code_extensions = ['py', 'js', 'html', 'css', 'java', 'c', 'cpp', 'cs', 'php', 'rb', 'go', 'rs', 'ts', 'sh', 'sql']
+            code_pattern = '|'.join(code_extensions)
+            code_url_match = re.search(f'https?://\\S+\\.({code_pattern})', question, re.IGNORECASE)
+            if code_url_match:
+                return self._read_code_file(code_url_match.group(0))
+
             # 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)
@@ -334,8 +371,7 @@ class BasicAgent:
                 
             
                 prompt = f"""
-                
-                               
+                   
                  
                 You: You are a general AI assistant. I will ask you a question. YOUR FINAL ANSWER should be a number OR as few words as possible OR a comma separated list of numbers and/or strings. If you are asked for a number, don't use comma to write your number neither use units such as $ or percent sign unless specified otherwise. If you are asked for a string, don't use articles, neither abbreviations (e.g. for cities), and write the digits in plain text unless specified otherwise. If you are asked for a comma separated list, apply the above rules depending of whether the element to be put in the list is a number or a string.            
                 Instructions: