Update app.py

app.py

@@ -2,7 +2,8 @@ import gradio as gr
 from gradio_client import Client
 import os
 import logging
-from urllib.parse import parse_qs, urlparse
+import sqlite3
+import requests
 
 # Initialize the client for image generation
 client_image = Client("mukaist/DALLE-4K")
@@ -21,6 +22,9 @@ DEFAULT_STYLE = "3840 x 2160"
 logging.basicConfig(level=logging.INFO)
 logger = logging.getLogger(__name__)
 
+# Initialize database
+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
@@ -45,12 +49,48 @@ def generate_image(prompt, resolution_key, style=DEFAULT_STYLE):
         logger.error(f"Error generating image: {e}")
         return None
 
-def gradio_interface(prompt, resolution_key):
+def update_user_points(username, 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))
+        conn.commit()
+    logger.info(f"User with username {username} had their points updated by {points}.")
+
+def notify_webhook(username):
+    """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
+    try:
+        response = requests.post(webhook_url, json=payload)
+        response.raise_for_status()
+        logger.info(f"Webhook notification sent successfully. Response: {response.status_code}")
+    except requests.RequestException as e:
+        logger.error(f"Error sending webhook notification: {e}")
+
+def get_user_points(username):
+    """Retrieve user points from the Flask server."""
+    webhook_url = "http://65.108.76.174:5000/get_points"
+    params = {'username': username}
+    try:
+        response = requests.get(webhook_url, params=params)
+        response.raise_for_status()
+        return response.json()
+    except requests.RequestException as e:
+        logger.error(f"Error fetching user points: {e}")
+        return {"error": "Failed to retrieve user points"}
+
+def gradio_interface(prompt, resolution_key, username):
     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
             return file_path, "The image was generated successfully."
         else:
             return None, "The image file is not available. Please try again later."
@@ -80,12 +120,21 @@ def create_gradio_interface(username):
         result_output = gr.Image(label="Generated Image", type="pil")
         message_output = gr.Textbox(label="Result", placeholder="Results will be shown here", interactive=False)
         
+        # Button to get user points
+        get_points_button = gr.Button("Get Points")
+        points_output = gr.Textbox(label="User Points", placeholder="User points will be displayed here", interactive=False)
+        
         # Set up interaction
         generate_button.click(
-            fn=lambda prompt, resolution_key: gradio_interface(prompt, resolution_key),
+            fn=lambda prompt, resolution_key: gradio_interface(prompt, resolution_key, username),
             inputs=[prompt_input, resolution_dropdown],
             outputs=[result_output, message_output]
         )
+
+        get_points_button.click(
+            fn=lambda: get_user_points(username),
+            outputs=points_output
+        )
         
         gr.HTML("""
         <style>
@@ -101,19 +150,6 @@ def launch_gradio(username):
     interface = create_gradio_interface(username)
     interface.launch()
 
-def extract_username_from_url(url):
-    parsed_url = urlparse(url)
-    params = parse_qs(parsed_url.query)
-    logger.debug(f"Parsed URL parameters: {params}")  # Debugging information
-    return params.get('username', ["Guest"])[0]
-
-def get_username_from_url(url):
-    # Extract username from the actual URL
-    username = extract_username_from_url(url)
-    logger.info(f"Extracted username: {username}")
-    return username
-
-# Replace this URL with your actual URL where the Gradio app is hosted
-url = 'https://measmonysuon-imagen.hf.space/?username=Guest'  # This should be dynamically obtained if needed
-username = get_username_from_url(url)
-launch_gradio(username)
+if __name__ == '__main__':
+    # Launch Gradio with a default username
+    launch_gradio("Guest")