Update app.py

app.py

@@ -2,7 +2,6 @@ import gradio as gr
 from gradio_client import Client
 import os
 import logging
-import sqlite3
 import requests
 from urllib.parse import urlparse, parse_qs
 
@@ -51,20 +50,24 @@ def generate_image(prompt, resolution_key, style=DEFAULT_STYLE):
         return None
 
 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 (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 ID {user_chat_id} had their points updated by {points}.")
+    """Updates the user's points using a webhook."""
+    webhook_url = f"{webhook_server}/update_points"
+    payload = {
+        'user_chat_id': user_chat_id,
+        'points': points
+    }
+    
+    try:
+        response = requests.post(webhook_url, json=payload)
+        response.raise_for_status()
+        logger.info(f"User with ID {user_chat_id} had their points updated by {points}. Webhook response: {response.status_code}")
+    except requests.RequestException as e:
+        logger.error(f"Error updating user points via webhook: {e}")
 
 def notify_webhook(user_chat_id):
     """Notify the webhook server about the points update."""
-    webhook_url = f"{webhook_server}/update_points"
-    payload = {'user_chat_id': user_chat_id, 'points': 10}  # Adjust payload as necessary
+    webhook_url = f"{webhook_server}/notify_update"
+    payload = {'user_chat_id': user_chat_id}
     try:
         response = requests.post(webhook_url, json=payload)
         response.raise_for_status()
@@ -87,7 +90,7 @@ def get_user_points(user_chat_id):
 def extract_user_chat_id_from_url(url):
     parsed_url = urlparse(url)
     params = parse_qs(parsed_url.query)
-    return params.get('user_chat_id', ["Guest"])[0]
+    return params.get('user_chat_id', ["Guest"])[0]]
 
 def gradio_interface(prompt, resolution_key, user_chat_id):
     result = generate_image(prompt, resolution_key)
@@ -103,7 +106,7 @@ def gradio_interface(prompt, resolution_key, user_chat_id):
     else:
         return None, "There was an error processing your photo. Please try again later."
 
-def create_gradio_interface():
+def create_gradio_interface(user_chat_id):
     with gr.Blocks() as interface:
         # Personalized HTML content
         gr.HTML("""
@@ -111,7 +114,7 @@ def create_gradio_interface():
         """)
 
         # Create input components
-        user_chat_id_input = gr.Textbox(label="User Chat ID", placeholder="Enter your user chat ID here...")
+        user_chat_id_input = gr.Textbox(label="User Chat ID", placeholder="Enter your user chat ID here...", value=user_chat_id)
         points_output = gr.Textbox(label="User Points", placeholder="User points will be displayed here", interactive=False)
         
         # Create the button to get user points
@@ -178,9 +181,11 @@ def create_gradio_interface():
     return interface
 
 def launch_gradio():
-    interface = create_gradio_interface()
+    # Dynamically obtain the URL, e.g., from an environment variable
+    url = os.getenv('USER_CHAT_ID_URL', 'http://example.com?user_chat_id=Guest')
+    user_chat_id = extract_user_chat_id_from_url(url)
+    interface = create_gradio_interface(user_chat_id)
     interface.launch()
 
 if __name__ == '__main__':
-    # Launch Gradio
     launch_gradio()