Update crew.py

crew.py

@@ -3,7 +3,7 @@
 # https://docs.crewai.com/introduction
 # https://ai.google.dev/gemini-api/docs
 
-import base64, os
+import os
 import pandas as pd
 from crewai import Agent, Crew, Process, Task
 from crewai.tools import tool
@@ -87,21 +87,16 @@ def run_crew(question, file_path):
            Raises:
                RuntimeError: If processing fails"""
         try:
-            client = OpenAI()
-    
-            img_b64 = ""
+            client = genai.Client(api_key=os.environ["GEMINI_API_KEY"])
             
-            with open(file_path, "rb") as file:
-                img_b64 = base64.b64encode(file.read()).decode("utf-8")
-    
-            completion = client.chat.completions.create(
-                model = "gpt-4.5-preview",
-                messages = [{"role": "user",
-                             "content": [{"type": "text", "text": question},
-                                         {"type": "image_url", "image_url": {"url": f"data:image/jpeg;base64,{img_b64}"}}]}]
+            file = client.files.upload(file=file_path)
+
+            response = client.models.generate_content(
+                model=IMAGE_ANALYSIS_MODEL,
+                contents=[file, question]
             )
-        
-            content = completion.choices[0].message.content
+          
+            return response.text
         except Exception as e:
             raise RuntimeError(f"Processing failed: {str(e)}")
 
@@ -265,7 +260,8 @@ def run_crew(question, file_path):
         goal="Given a question and image file, analyze the image and answer the question: {question}",
         backstory="As an expert image analysis assistant, you analyze the image to answer the question.",
         allow_delegation=False,
-        llm=AGENT_MODEL,
+        #llm=AGENT_MODEL,
+        llm=IMAGE_ANALYSIS_MODEL,
         max_iter=2,
         tools=[image_analysis_tool],
         verbose=True