Update app.py

app.py

@@ -82,10 +82,14 @@ def get_user_points(user_chat_id):
     try:
         response = requests.get(webhook_url, params=params)
         response.raise_for_status()
-        return response.json().get('points', 0)  # Default to 0 if points not found
+        data = response.json()
+        if 'points' in data:
+            return data['points']
+        else:
+            return None  # User not found or points not in response
     except requests.RequestException as e:
         logger.error(f"Error fetching user points: {e}")
-        return 0
+        return None
 
 def extract_user_chat_id_from_url(url):
     parsed_url = urlparse(url)
@@ -141,6 +145,12 @@ def create_gradio_interface(user_chat_id):
         
         def handle_generate_image(prompt, resolution_key, user_chat_id):
             points = get_user_points(user_chat_id)
+            
+            # Check if user_chat_id is found in the database
+            if points is None:
+                return None, "User not found. Please register with @myexpsupportBot."
+        
+            # Check if the user has enough points
             if points >= 5:
                 result = gradio_interface(prompt, resolution_key, user_chat_id)
                 if result[0]:  # If image generation is successful