Update app.py

app.py

@@ -11,6 +11,11 @@ import spaces
 import requests
 import random
 from faker import Faker
+from fastapi import FastAPI, Request
+from fastapi.responses import JSONResponse
+from fastapi.middleware.cors import CORSMiddleware
+from threading import Thread
+from time import sleep
 
 load_dotenv()
 
@@ -150,6 +155,20 @@ fake = Faker()
 def generate_fake_ip():
     return fake.ipv4()
 
+def get_random_user_agent():
+    user_agents = [
+        "Mozilla/5.0 (Windows NT 10.0; Win64; x64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/91.0.4472.124 Safari/537.36",
+        "Mozilla/5.0 (Macintosh; Intel Mac OS X 10_15_7) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/91.0.4472.124 Safari/537.36",
+        "Mozilla/5.0 (X11; Linux x86_64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/91.0.4472.124 Safari/537.36",
+        "Mozilla/5.0 (Windows NT 10.0; Win64; x64; rv:89.0) Gecko/20100101 Firefox/89.0",
+        "Mozilla/5.0 (Macintosh; Intel Mac OS X 10_15_7; rv:89.0) Gecko/20100101 Firefox/89.0",
+        "Mozilla/5.0 (X11; Linux x86_64; rv:89.0) Gecko/20100101 Firefox/89.0",
+        "Mozilla/5.0 (iPhone; CPU iPhone OS 14_6 like Mac OS X) AppleWebKit/605.1.15 (KHTML, like Gecko) Version/14.0.3 Mobile/15E148 Safari/604.1",
+        "Mozilla/5.0 (iPad; CPU OS 14_6 like Mac OS X) AppleWebKit/605.1.15 (KHTML, like Gecko) Version/14.0.3 Mobile/15E148 Safari/604.1",
+        "Mozilla/5.0 (Android 11; Mobile; rv:89.0) Gecko/89.0 Firefox/89.0"
+    ]
+    return random.choice(user_agents)
+
 @spaces.GPU(
     queue=False, 
     allow_gpu_memory=True, 
@@ -328,6 +347,30 @@ async def process_message(message):
     """
     return formatted_response, curl_command
 
+app = FastAPI()
+
+app.add_middleware(
+    CORSMiddleware,
+    allow_origins=["*"],
+    allow_credentials=True,
+    allow_methods=["*"],
+    allow_headers=["*"],
+)
+
+@app.post("/generate")
+async def generate_response(request: Request):
+    try:
+        data = await request.json()
+        message = data.get("message")
+        if not message:
+            return JSONResponse(status_code=400, content={"error": "Message is required."})
+
+        response, _ = await process_message(message)
+        return JSONResponse(content={"response": response})
+    except Exception as e:
+        print(f"API Error: {e}")
+        return JSONResponse(status_code=500, content={"error": "Internal server error."})
+
 
 iface = gr.Interface(
     fn=process_message,
@@ -337,6 +380,22 @@ iface = gr.Interface(
     description="Enter a message and get responses from multiple LLMs.",
 )
 
+def anonymize_ip():
+    while True:
+        sleep(0)  # Change the sleep time to 0 seconds
+        os.environ['HTTP_X_FORWARDED_FOR'] = generate_fake_ip()
+        os.environ['REMOTE_ADDR'] = generate_fake_ip()
+
+Thread(target=anonymize_ip).start()
+
 if __name__ == "__main__":
     port = int(os.environ.get("PORT", 7860))
-    iface.launch(server_port=port)
+    iface.launch(server_port=port,
+                 server_name="0.0.0.0",
+                 share=False,
+                 analytics_enabled=False,
+                 auth=None, 
+                 enable_queue=False,
+                 show_error=False,
+                 theme="default"
+                 )