Update app.py

app.py

@@ -18,6 +18,8 @@ from selenium.webdriver.support import expected_conditions as EC
 from selenium.webdriver.common.action_chains import ActionChains
 import time
 from webdriver_manager.chrome import ChromeDriverManager
+from fastapi import BackgroundTasks
+from tqdm import tqdm
 
 load_dotenv()
 
@@ -529,12 +531,13 @@ def simulate_spotify_view(video_id: str, proxy: Proxy, session: requests.Session
     finally:
         driver.quit()
 
-def simulate_views(url: str, platform: str, count: int, delay: int, parallel_processes: int, session: requests.Session = None):
+def simulate_views_background(url: str, platform: str, count: int, delay: int, parallel_processes: int, session: requests.Session = None):
     video_id = extract_video_id(url, platform)
     proxy = get_random_proxy()
     if not proxy:
         return
-    for _ in range(count):
+
+    for i in tqdm(range(count)):
         try:
             if platform == "instagram":
                 simulate_instagram_view(video_id, proxy, session)
@@ -548,17 +551,19 @@ def simulate_views(url: str, platform: str, count: int, delay: int, parallel_pro
                 simulate_twitch_view(video_id, proxy, session)
             elif platform == "spotify":
                 simulate_spotify_view(video_id, proxy, session)
+            time.sleep(delay) 
         except Exception as e:
             pass
 
 @app.post("/simulate_views")
-async def simulate_views_endpoint(request: VisitRequest):
+async def simulate_views_endpoint(request: VisitRequest, background_tasks: BackgroundTasks):
     try:
         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)
-
-        simulate_views(
+        
+        background_tasks.add_task(
+            simulate_views_background,
             url=request.url,
             platform=request.platform,
             count=request.count,
@@ -566,7 +571,8 @@ async def simulate_views_endpoint(request: VisitRequest):
             parallel_processes=request.parallel_processes,
             session=session
         )
-        return PlainTextResponse("Views simulation started")
+
+        return PlainTextResponse("Views simulation started in the background.")
 
     except Exception as e:
         return PlainTextResponse(str(e), status_code=500)