Update app.py

app.py

@@ -76,20 +76,15 @@ def notify_webhook(user_chat_id):
         logger.error(f"Error sending webhook notification: {e}")
 
 def get_user_points(user_chat_id):
-    """Retrieve user points from the Flask server."""
-    webhook_url = f"{webhook_server}/get_points"
-    params = {'user_chat_id': user_chat_id}
-    try:
-        response = requests.get(webhook_url, params=params)
-        response.raise_for_status()
-        data = response.json()
-        if 'points' in data:
-            return data['points']
+    """Fetches user points from the database. Returns 'User not found' if the user is not in the database."""
+    with sqlite3.connect(db_path) as conn:
+        cursor = conn.cursor()
+        cursor.execute('SELECT points FROM users WHERE user_chat_id = ?', (user_chat_id,))
+        result = cursor.fetchone()
+        if result:
+            return result[0]  # Return the 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 None
+            return "User not found"  # Return an error message if user is not found
 
 def extract_user_chat_id_from_url(url):
     parsed_url = urlparse(url)
@@ -110,6 +105,23 @@ def gradio_interface(prompt, resolution_key, user_chat_id):
     else:
         return None, "There was an error processing your photo. Please try again later."
 
+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"
+    
+    if points >= 5:
+        result = gradio_interface(prompt, resolution_key, user_chat_id)
+        if result[0]:  # If image generation is successful
+            return result[0], "The image was generated successfully."
+        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."        
+        
+
 def create_gradio_interface(user_chat_id):
     with gr.Blocks() as interface:
         # Personalized HTML content
@@ -118,12 +130,14 @@ def create_gradio_interface(user_chat_id):
         """)
 
         # Create input components
-        user_chat_id_input = gr.Textbox(label="User Chat ID", placeholder="Enter your user chat ID here...", value=user_chat_id, interactive=False)  # Make the textbox non-editable
+        user_chat_id_input = gr.Textbox(
+            label="User Chat ID",
+            placeholder="Enter your user chat ID here...",
+            value=user_chat_id,
+            interactive=False  # Make the textbox non-editable
+        )
         points_output = gr.Textbox(label="User Points", placeholder="User points will be displayed here", interactive=False)
         
-        # Create the button to get user points
-        get_points_button = gr.Button("Get Points")
-        
         # Create other components
         prompt_input = gr.Textbox(label="Prompt", placeholder="Enter your prompt here...")
         resolution_dropdown = gr.Dropdown(choices=list(resolutions.keys()), label="Resolution", value="1024x1024")
@@ -136,44 +150,19 @@ def create_gradio_interface(user_chat_id):
         with gr.Row():
             user_chat_id_input
             points_output
-            get_points_button
         
         with gr.Row():
             prompt_input
             resolution_dropdown
             generate_button
         
-        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
-                    return result[0], "The image was generated successfully."
-                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."
-
         # Set up interactions
         generate_button.click(
             fn=handle_generate_image,
             inputs=[prompt_input, resolution_dropdown, user_chat_id_input],
             outputs=[result_output, message_output]
         )
-        
-        # Remove the get_points_button related interaction if it's not needed anymore
-        get_points_button.click(
-            fn=lambda user_chat_id: get_user_points(user_chat_id),
-            inputs=[user_chat_id_input],
-            outputs=points_output
-        )
-        
+
         gr.HTML("""
         <style>
             footer.svelte-1rjryqp {