Update app.py

app.py

@@ -20,6 +20,7 @@ import time
 from webdriver_manager.chrome import ChromeDriverManager
 from fastapi import BackgroundTasks
 from tqdm import tqdm
+import asyncio
 
 load_dotenv()
 
@@ -531,13 +532,18 @@ def simulate_spotify_view(video_id: str, proxy: Proxy, session: requests.Session
     finally:
         driver.quit()
 
+
 def simulate_views_background(url: str, platform: str, count: int, delay: int, parallel_processes: int, session: requests.Session = None):
+    for _ in range(parallel_processes): 
+        asyncio.run(simulate_one_view_process(url, platform, count, delay, session))
+
+async def simulate_one_view_process(url: str, platform: str, count: int, delay: int, session: requests.Session = None):
     video_id = extract_video_id(url, platform)
     proxy = get_random_proxy()
     if not proxy:
         return
 
-    for i in tqdm(range(count)):
+    for i in tqdm(range(count), desc=f"Simulating views for {url}"):
         try:
             if platform == "instagram":
                 simulate_instagram_view(video_id, proxy, session)
@@ -551,7 +557,7 @@ def simulate_views_background(url: str, platform: str, count: int, delay: int, p
                 simulate_twitch_view(video_id, proxy, session)
             elif platform == "spotify":
                 simulate_spotify_view(video_id, proxy, session)
-            time.sleep(delay) 
+            await asyncio.sleep(delay) 
         except Exception as e:
             pass
 
@@ -561,7 +567,7 @@ async def simulate_views_endpoint(request: VisitRequest, background_tasks: Backg
         session = None
         if os.getenv(f'{request.platform.upper()}_USER') and os.getenv(f'{request.platform.upper()}_PASSWORD'):
             session = authenticate(os.getenv(f'{request.platform.upper()}_USER'), os.getenv(f'{request.platform.upper()}_PASSWORD'), request.platform)
-        
+
         background_tasks.add_task(
             simulate_views_background,
             url=request.url,
@@ -602,18 +608,23 @@ async def login(username: str = Form(...), password: str = Form(...), platform:
         return e
 
 @app.post("/simulate")
-async def simulate(urls: str = Form(...), platform: str = Form(...), count: int = Form(...), delay: int = Form(...), parallel_processes: int = Form(...)):
+async def simulate(urls: str = Form(...), platform: str = Form(...), count: int = Form(...), delay: int = Form(...), parallel_processes: int = Form(...), background_tasks: BackgroundTasks):
     try:
         session = None
         if os.getenv(f'{platform.upper()}_USER') and os.getenv(f'{platform.upper()}_PASSWORD'):
             session = authenticate(os.getenv(f'{platform.upper()}_USER'), os.getenv(f'{platform.upper()}_PASSWORD'), platform)
 
         for index, url in enumerate(urls.split("\n")):
-            try:
-                simulate_views(url, platform, count, delay, parallel_processes, session)
-            except Exception as e:
-                pass
-        return PlainTextResponse("Simulation completed.")
+            background_tasks.add_task(
+                simulate_views_background,
+                url=url,
+                platform=platform,
+                count=count,
+                delay=delay,
+                parallel_processes=parallel_processes,
+                session=session
+            )
+        return PlainTextResponse("Simulations started in the background.")
 
     except Exception as e:
         return PlainTextResponse(str(e), status_code=500)