Update app.py

app.py

@@ -23,9 +23,6 @@ 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
@@ -63,7 +60,7 @@ def update_user_points(username, points):
 
 def notify_webhook(username):
     """Notify the webhook server about the points update."""
-    webhook_url = "http://65.108.76.174:5000/update_points"
+    webhook_url = f"{webhook_server}/update_points"
     payload = {'username': username, 'points': 10}  # Adjust payload as necessary
     try:
         response = requests.post(webhook_url, json=payload)
@@ -74,7 +71,7 @@ def notify_webhook(username):
 
 def get_user_points(username):
     """Retrieve user points from the Flask server."""
-    webhook_url = "http://65.108.76.174:5000/get_points"
+    webhook_url = f"{webhook_server}/get_points"
     params = {'username': username}
     try:
         response = requests.get(webhook_url, params=params)
@@ -89,8 +86,7 @@ def extract_username_from_url(url):
     params = parse_qs(parsed_url.query)
     return params.get('username', ["Guest"])[0]
 
-def gradio_interface(prompt, resolution_key, url):
-    username = extract_username_from_url(url)
+def gradio_interface(prompt, resolution_key, username):
     result = generate_image(prompt, resolution_key)
     
     if result and result[0]:
@@ -135,8 +131,8 @@ def create_gradio_interface():
         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, default="Guest", elem_id="username-input")
-        
+        username_input = gr.Textbox(visible=False, default="Guest")
+
         # Set up interaction
         generate_button.click(
             fn=lambda prompt, resolution_key, username: gradio_interface(prompt, resolution_key, username),