Update app.py

app.py

@@ -96,6 +96,8 @@ def cached_search(query):
 # ----- THIS IS WERE YOU CAN BUILD WHAT YOU WANT ------
 
 
+
+        
 class BasicAgent:
     
     
@@ -155,10 +157,25 @@ class BasicAgent:
             traceback.print_exc()  
             return f"I apologize, but I encountered an error while processing your question. Error: {str(e)}"
 
-
+    def _process_image_query(self, question, image_url):
+        try:
+            response = requests.get(image_url)
+            image_data = response.content
+            multimodal_prompt = [{"text": question}, {"image": {"data": image_data}}]
+            response = self.gemini_model.generate_content(multimodal_prompt)
+            return response.text
+        except Exception as e:
+            print(f"Error processing image: {str(e)}")
+            return f"I encountered an error analyzing the image: {str(e)}"
     
     def process_question(self, question: str) -> str:
         try:
+            # Processing image
+            image_url_match = re.search(r'https?://\S+\.(jpg|jpeg|png|gif|webp)', question, re.IGNORECASE)
+            if image_url_match and self.gemini_model:
+                image_url = image_url_match.group(0)
+                return self._process_image_query(question, image_url)
+                
             # Check if this is a request about a YouTube video
             youtube_patterns = ["youtube.com", "youtu.be", "watch youtube", "youtube video"]
             use_youtube_tool = any(pattern in question.lower() for pattern in youtube_patterns)