Update app.py

app.py

@@ -8,6 +8,10 @@ import gradio as gr
 
 # Set your Hugging Face API token
 HUGGING_FACE_API_KEY = os.getenv("HUGGING_FACE_API_KEY")
+
+if not HUGGING_FACE_API_KEY:
+    raise ValueError("Please set the Hugging Face API key in the environment as 'HUGGING_FACE_API_KEY'.")
+
 math_messages = []
 
 # Function to process the image with Hugging Face API
@@ -36,8 +40,12 @@ def process_image(image, shouldConvert=False):
             files={"file": img_file}
         )
     os.remove(filename)  # Clean up temp file
-    
-    caption = response.json().get("generated_text", "No description available.")
+
+    # Check if response is successful and handle errors
+    if response.status_code == 200:
+        caption = response.json().get("generated_text", "No description available.")
+    else:
+        caption = f"Error: {response.status_code} - {response.json().get('error', 'Unknown error')}"
     return caption
 
 # Function for getting math responses from Hugging Face's text generation API
@@ -61,8 +69,13 @@ def get_math_response(image_description, user_question):
         headers={"Authorization": f"Bearer {HUGGING_FACE_API_KEY}"},
         json=payload
     )
-    
-    answer = response.json().get("generated_text", "Sorry, I couldn't generate a response.")
+
+    # Check if response is successful and handle errors
+    if response.status_code == 200:
+        answer = response.json().get("generated_text", "Sorry, I couldn't generate a response.")
+    else:
+        answer = f"Error: {response.status_code} - {response.json().get('error', 'Unknown error')}"
+
     yield answer
     math_messages.append({"role": "assistant", "content": answer})