Update app.py

app.py

@@ -1,8 +1,41 @@
-import subprocess
+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)
 
 # Initialize the client for image generation
 client_image = Client("mukaist/DALLE-4K")
@@ -21,6 +54,10 @@ 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
@@ -87,14 +124,40 @@ def create_gradio_interface(username):
 def launch_gradio(username=None):
     if not username:
         username = "Guest"
-    
     # Launch the Gradio interface
     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__":
-    # Start the Gradio interface
-    launch_gradio(username="Guest")
+    # Launch Gradio in a separate thread
+    gradio_thread = threading.Thread(target=lambda: launch_gradio("Guest"))
+    gradio_thread.start()
 
-    # Start the Telegram bot
-    subprocess.Popen(["python", "telegrabot.py"])
+    # Start Telegram bot polling
+    run_telegram_bot()