Update app.py

app.py

@@ -11,21 +11,34 @@ client = None
 def initialize_client(api_key=None):
     global client
     api_key = os.getenv("TOGETHER_API_KEY")
-    client = Together()
+    if api_key:
+        client = Together()
+        print("Client initialized successfully.")
+    else:
+        print("API key not found. Please set TOGETHER_API_KEY environment variable.")
 
 def encode_image(image_path):
-    with Image.open(image_path) as img:
-        buffered = io.BytesIO()
-        img.save(buffered, format="PNG")
-        return base64.b64encode(buffered.getvalue()).decode("utf-8")
+    try:
+        with Image.open(image_path) as img:
+            buffered = io.BytesIO()
+            img.save(buffered, format="PNG")
+            return base64.b64encode(buffered.getvalue()).decode("utf-8")
+    except Exception as e:
+        print(f"Error encoding image: {str(e)}")
+        return None
 
 def detect_accident(image_path):
     if client is None:
         initialize_client()
+        if client is None:
+            return "Error: Client not initialized."
 
     prompt = "Determine if the radiator in the image shows signs of being involved in an accident or not."
     image_base64 = encode_image(image_path)
     
+    if not image_base64:
+        return "Error: Could not encode the image. Please try again with a different one."
+
     messages = [
         {"role": "system", "content": prompt},
         {
@@ -43,12 +56,13 @@ def detect_accident(image_path):
             max_tokens=10,
             temperature=0.5
         ).choices[0].message["content"].strip()
-
-        # Limit the response to a simple "accident" or "not accident" output
+        
+        # Check if response contains "accident" or "not accident"
         return "accident" if "accident" in response.lower() else "not accident"
 
     except Exception as e:
-        return "Error: Could not process the image. Please try again with a different one."
+        print(f"Error during model call: {str(e)}")
+        return f"Error: Could not process the image. Please try again with a different one."
 
 # Gradio interface
 with gr.Blocks() as demo: