Update app.py

app.py

@@ -121,9 +121,18 @@ def gradio_interface(prompt, resolution_key, user_chat_id):
 def handle_generate_image(prompt, resolution_key, user_chat_id):
     points = get_user_points(user_chat_id)
     
-    if points == "User not found":
-        # If user is not found, update points_output with the registration message
-        return None, "Please register with @myexpsupportBot"
+    # Check if the points is a string indicating user not found
+    if isinstance(points, str):
+        if points == "User not found":
+            return None, "Please register with @myexpsupportBot"
+        else:
+            return None, "Error retrieving points. Please try again later."
+    
+    # Ensure points is an integer
+    try:
+        points = int(points)
+    except ValueError:
+        return None, "Error processing points. Please try again later."
     
     if points >= 5:
         result = gradio_interface(prompt, resolution_key, user_chat_id)
@@ -132,7 +141,7 @@ def handle_generate_image(prompt, resolution_key, user_chat_id):
         else:
             return None, "There was an error processing your photo. Please try again later."
     else:
-        return None, "Insufficient points. Please get more points before generating an image."        
+        return None, "Insufficient points. Please get more points before generating an image."       
         
 
 def create_gradio_interface():