Update app.py

app.py

@@ -1,41 +1,8 @@
-import telebot
 import gradio as gr
 from gradio_client import Client
 import os
 import logging
-import socks
-import socket
-import requests
-from requests.adapters import HTTPAdapter
-from requests.packages.urllib3.util.retry import Retry
-import threading
-
-# Proxy configuration
-PROXY_HOST = '64.29.87.81'
-PROXY_PORT = 48558
-PROXY_USERNAME = 'KNaDX4iRd2t6El0'
-PROXY_PASSWORD = 'NleyoQHQJwHbqSH'
-
-# Configure SOCKS proxy
-socks.set_default_proxy(socks.SOCKS5, PROXY_HOST, PROXY_PORT, username=PROXY_USERNAME, password=PROXY_PASSWORD)
-socket.socket = socks.socksocket
-
-# Create a custom requests session with proxy
-session = requests.Session()
-session.proxies.update({
-    'http': f'socks5://{PROXY_USERNAME}:{PROXY_PASSWORD}@{PROXY_HOST}:{PROXY_PORT}',
-    'https': f'socks5://{PROXY_USERNAME}:{PROXY_PASSWORD}@{PROXY_HOST}:{PROXY_PORT}',
-})
-
-# Add retry strategy to the session
-retry_strategy = Retry(
-    total=5,
-    backoff_factor=1,
-    status_forcelist=[500, 502, 503, 504]
-)
-adapter = HTTPAdapter(max_retries=retry_strategy)
-session.mount('http://', adapter)
-session.mount('https://', adapter)
+from urllib.parse import parse_qs, urlparse
 
 # Initialize the client for image generation
 client_image = Client("mukaist/DALLE-4K")
@@ -54,10 +21,6 @@ DEFAULT_STYLE = "3840 x 2160"
 logging.basicConfig(level=logging.INFO)
 logger = logging.getLogger(__name__)
 
-# Initialize the bot with your API token
-API_TOKEN = 'YOUR_TELEGRAM_BOT_API_TOKEN'
-bot = telebot.TeleBot(API_TOKEN)
-
 def generate_image(prompt, resolution_key, style=DEFAULT_STYLE):
     resolution = resolutions.get(resolution_key, (1024, 1024))
     width, height = resolution
@@ -93,6 +56,11 @@ def gradio_interface(prompt, resolution_key):
     else:
         return None, "There was an error processing your photo. Please try again later."
 
+def extract_username_from_url(url):
+    parsed_url = urlparse(url)
+    params = parse_qs(parsed_url.query)
+    return params.get('username', ["Guest"])[0]
+
 def create_gradio_interface(username):
     description = f"Welcome, {username}! Generate images based on prompts."
 
@@ -128,36 +96,13 @@ def launch_gradio(username=None):
     interface = create_gradio_interface(username)
     interface.launch()
 
-@bot.message_handler(commands=['start'])
-def handle_start(message):
-    # Extract username from message
-    username = message.from_user.username or "Guest"
-    
-    # Construct URL with username
-    encoded_url = f"https://measmonysuon-webapp.hf.space/?username={username}"
-    
-    # Send a message with the inline button to open the web app
-    markup = telebot.types.InlineKeyboardMarkup()
-    button = telebot.types.InlineKeyboardButton(
-        text="Open Web App",
-        web_app=telebot.types.WebAppInfo(url=encoded_url)
-    )
-    markup.add(button)
-    
-    bot.send_message(
-        message.chat.id,
-        f"Welcome, {username}! Click the button below to open the web app:",
-        reply_markup=markup
-    )
-
-def run_telegram_bot():
-    bot.polling()
-
-# Run Gradio and Telegram bot concurrently
-if __name__ == "__main__":
-    # Launch Gradio in a separate thread
-    gradio_thread = threading.Thread(target=lambda: launch_gradio("Guest"))
-    gradio_thread.start()
+def get_username_from_url():
+    # Placeholder URL, replace with actual URL or method to get URL in deployment
+    url = 'https://measmonysuon-webapp.hf.space/?username=MoMo'  
+    username = extract_username_from_url(url)
+    logger.info(f"Extracted username: {username}")
+    return username
 
-    # Start Telegram bot polling
-    run_telegram_bot()
+# Launch Gradio with the extracted or default username
+username = get_username_from_url()
+launch_gradio(username)