Update utils/basic_functions.py

utils/basic_functions.py

@@ -76,8 +76,6 @@ def request_manager(model_name,url):
 def gemini_identify_id(model_name,image,system_prompt):
     # 2. Function to process image with Gemini Pro Vision
     try:
-        image = get_image(url)
-        
         genai.configure(api_key=get_secret('GEMINI_API'))
 
         model = genai.GenerativeModel("gemini-2.0-flash")
@@ -85,11 +83,11 @@ def gemini_identify_id(model_name,image,system_prompt):
         response_text = response.text
         if not response_text:
             print('Could not find an ID number')
-            exit()
-        print(response_text)
+            return [image,'no response was received']
+        print(f"response_text: {response_text}")
     
     except Exception as e:
-        return f"Error processing image: {str(e)}",None
+        return [image,f"Error processing image: {str(e)}"]
     
     draw = ImageDraw.Draw(image)
     draw.rectangle(ast.literal_eval(response_text),outline='yellow',width=5)