Update app.py

app.py

@@ -529,7 +529,7 @@ 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):
+def simulate_views(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:
@@ -554,19 +554,20 @@ def simulate_views(url: str, platform: str, count: int, delay: int, parallel_pro
 @app.post("/simulate_views")
 async def simulate_views_endpoint(request: VisitRequest):
     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(
-                url=request.url,
-                platform=request.platform,
-                count=request.count,
-                delay=request.delay,
-                parallel_processes=request.parallel_processes,
-                session=session
-            )
-            return PlainTextResponse("Views simulation started")
-        else:
-            return PlainTextResponse("Credentials not found for this platform.")
+
+        simulate_views(
+            url=request.url,
+            platform=request.platform,
+            count=request.count,
+            delay=request.delay,
+            parallel_processes=request.parallel_processes,
+            session=session
+        )
+        return PlainTextResponse("Views simulation started")
+
     except Exception as e:
         return PlainTextResponse(str(e), status_code=500)
 
@@ -597,16 +598,17 @@ async def login(username: str = Form(...), password: str = Form(...), platform:
 @app.post("/simulate")
 async def simulate(urls: str = Form(...), platform: str = Form(...), count: int = Form(...), delay: int = Form(...), parallel_processes: int = Form(...)):
     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.")
-        else:
-            return PlainTextResponse("Credentials not found for this 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.")
+
     except Exception as e:
         return PlainTextResponse(str(e), status_code=500)