Update app.py

app.py

@@ -23,6 +23,9 @@ DEFAULT_STYLE = "3840 x 2160"
 logging.basicConfig(level=logging.INFO)
 logger = logging.getLogger(__name__)
 
+# Database path
+db_path = 'C:/Users/Administrator/Desktop/TGBoot/photo2prompt_db/users.db'
+
 def generate_image(prompt, resolution_key, style=DEFAULT_STYLE):
     resolution = resolutions.get(resolution_key, (1024, 1024))
     width, height = resolution
@@ -47,21 +50,21 @@ def generate_image(prompt, resolution_key, style=DEFAULT_STYLE):
         logger.error(f"Error generating image: {e}")
         return None
 
-def update_user_points(username, points):
+def update_user_points(user_chat_id, points):
     """Updates the user's points in the database."""
     with sqlite3.connect(db_path) as conn:
         cursor = conn.cursor()
         cursor.execute('''
-        INSERT INTO users (username, points) VALUES (?, ?)
-        ON CONFLICT(username) DO UPDATE SET points = points + ?
-        ''', (username, points, points))
+        INSERT INTO users (user_chat_id, points) VALUES (?, ?)
+        ON CONFLICT(user_chat_id) DO UPDATE SET points = points + ?
+        ''', (user_chat_id, points, points))
         conn.commit()
-    logger.info(f"User with username {username} had their points updated by {points}.")
+    logger.info(f"User with user_chat_id {user_chat_id} had their points updated by {points}.")
 
-def notify_webhook(username):
+def notify_webhook(user_chat_id):
     """Notify the webhook server about the points update."""
     webhook_url = "http://65.108.76.174:5000/update_points"
-    payload = {'username': username, 'points': 10}  # Adjust payload as necessary
+    payload = {'user_chat_id': user_chat_id, 'points': 10}  # Adjust payload as necessary
     try:
         response = requests.post(webhook_url, json=payload)
         response.raise_for_status()
@@ -69,10 +72,10 @@ def notify_webhook(username):
     except requests.RequestException as e:
         logger.error(f"Error sending webhook notification: {e}")
 
-def get_user_points(username):
+def get_user_points(user_chat_id):
     """Retrieve user points from the Flask server."""
     webhook_url = "http://65.108.76.174:5000/get_points"
-    params = {'username': username}
+    params = {'user_chat_id': user_chat_id}
     try:
         response = requests.get(webhook_url, params=params)
         response.raise_for_status()
@@ -81,19 +84,19 @@ def get_user_points(username):
         logger.error(f"Error fetching user points: {e}")
         return "Failed to retrieve user points"
 
-def extract_username_from_url(url):
+def extract_user_chat_id_from_url(url):
     parsed_url = urlparse(url)
     params = parse_qs(parsed_url.query)
-    return params.get('username', ["Guest"])[0]
+    return params.get('user_chat_id', ["Guest"])[0]
 
-def gradio_interface(prompt, resolution_key, username):
+def gradio_interface(prompt, resolution_key, user_chat_id):
     result = generate_image(prompt, resolution_key)
     
     if result and result[0]:
         file_path = result[0][0].get('image')
         if file_path and os.path.exists(file_path):
-            update_user_points(username, 10)  # Award 10 points for each image generated
-            notify_webhook(username)  # Notify the webhook server
+            update_user_points(user_chat_id, 10)  # Award 10 points for each image generated
+            notify_webhook(user_chat_id)  # Notify the webhook server
             return file_path, "The image was generated successfully."
         else:
             return None, "The image file is not available. Please try again later."
@@ -130,19 +133,19 @@ def create_gradio_interface():
         get_points_button = gr.Button("Get Points")
         points_output = gr.Textbox(label="User Points", placeholder="User points will be displayed here", interactive=False)
         
-        # Hidden textbox to hold the username
-        username_input = gr.Textbox(visible=False, value="Guest", elem_id="username-input")
+        # Hidden textbox to hold the user_chat_id
+        user_chat_id_input = gr.Textbox(visible=False, value="Guest", elem_id="username-input")
 
         # Set up interaction
         generate_button.click(
-            fn=lambda prompt, resolution_key, username: gradio_interface(prompt, resolution_key, username),
-            inputs=[prompt_input, resolution_dropdown, username_input],
+            fn=lambda prompt, resolution_key, user_chat_id: gradio_interface(prompt, resolution_key, user_chat_id),
+            inputs=[prompt_input, resolution_dropdown, user_chat_id_input],
             outputs=[result_output, message_output]
         )
 
         get_points_button.click(
-            fn=lambda username: get_user_points(username),
-            inputs=[username_input],
+            fn=lambda user_chat_id: get_user_points(user_chat_id),
+            inputs=[user_chat_id_input],
             outputs=points_output
         )