Update app.py

app.py

@@ -9,6 +9,26 @@ app = Flask(__name__)
 def generate_random_ip():
     return f"{random.randint(1,255)}.{random.randint(0,255)}.{random.randint(0,255)}.{random.randint(0,255)}"
 
+def generate_user_agent():
+    os_list = ['Windows NT 10.0', 'Windows NT 6.1', 'Mac OS X 10_15_7', 'Ubuntu', 'Linux x86_64']
+    browser_list = ['Chrome', 'Firefox', 'Safari', 'Edge']
+    chrome_version = f"{random.randint(70, 126)}.0.{random.randint(1000, 9999)}.{random.randint(100, 999)}"
+    firefox_version = f"{random.randint(70, 100)}.0"
+    safari_version = f"{random.randint(600, 615)}.{random.randint(1, 9)}.{random.randint(1, 9)}"
+    edge_version = f"{random.randint(80, 100)}.0.{random.randint(1000, 9999)}.{random.randint(100, 999)}"
+
+    os = random.choice(os_list)
+    browser = random.choice(browser_list)
+
+    if browser == 'Chrome':
+        return f"Mozilla/5.0 ({os}) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/{chrome_version} Safari/537.36"
+    elif browser == 'Firefox':
+        return f"Mozilla/5.0 ({os}; rv:{firefox_version}) Gecko/20100101 Firefox/{firefox_version}"
+    elif browser == 'Safari':
+        return f"Mozilla/5.0 (Macintosh; Intel Mac OS X 10_15_7) AppleWebKit/{safari_version} (KHTML, like Gecko) Version/{safari_version.split('.')[0]}.1.2 Safari/{safari_version}"
+    elif browser == 'Edge':
+        return f"Mozilla/5.0 ({os}) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/{edge_version} Safari/537.36 Edg/{edge_version}"
+        
 def format_openai_response(content, finish_reason=None):
     return {
         "id": "chatcmpl-123",
@@ -54,7 +74,7 @@ def chat_completions():
         'content-type': 'application/json',
         'X-Forwarded-For': generate_random_ip(),
         'origin': 'https://chatpro.ai-pro.org',
-        'user-agent': 'Mozilla/5.0 (Windows NT 10.0; Win64; x64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/127.0.0.0 Safari/537.36'
+        'user-agent': generate_user_agent()
     }
     
     payload = {