gemin mock added

services/gemini_service.py

@@ -31,6 +31,12 @@ DOWNLOADS_DIR = os.path.join(os.path.dirname(os.path.dirname(__file__)), "downlo
 # Ensure downloads directory exists
 os.makedirs(DOWNLOADS_DIR, exist_ok=True)
 
+# Mock mode for local testing (set GEMINI_MOCK_MODE=true to skip real API calls)
+MOCK_MODE = os.getenv("GEMINI_MOCK_MODE", "false").lower() == "true"
+
+# Sample video URL for mock mode (a public test video)
+MOCK_VIDEO_URL = "https://video.twimg.com/amplify_video/1994083297756848128/vid/avc1/576x576/ue31qU0xts8L9tXD.mp4?tag=21"
+
 # Concurrency limits from environment (defaults)
 MAX_CONCURRENT_VIDEOS = int(os.getenv("MAX_CONCURRENT_VIDEOS", "2"))
 MAX_CONCURRENT_IMAGES = int(os.getenv("MAX_CONCURRENT_IMAGES", "5"))
@@ -179,6 +185,17 @@ class GeminiService:
         Start video generation using Veo model.
         Returns operation details for polling.
         """
+        # Mock mode for testing without API credits
+        if MOCK_MODE:
+            import uuid
+            mock_operation_name = f"mock_operation_{uuid.uuid4().hex[:16]}"
+            logger.info(f"[MOCK MODE] Starting video generation: {mock_operation_name}")
+            return {
+                "gemini_operation_name": mock_operation_name,
+                "done": False,
+                "status": "pending"
+            }
+        
         async with get_video_semaphore():
             try:
                 # Start video generation
@@ -211,6 +228,19 @@ class GeminiService:
         Check the status of a video generation operation.
         Returns status and video URL if complete.
         """
+        # Mock mode for testing without API credits
+        if MOCK_MODE:
+            # Simulate processing time: complete after 2 checks (track via a simple mechanism)
+            # For simplicity, always return completed with mock video URL
+            logger.info(f"[MOCK MODE] Checking video status: {gemini_operation_name}")
+            await asyncio.sleep(2)  # Simulate API delay
+            return {
+                "gemini_operation_name": gemini_operation_name,
+                "done": True,
+                "status": "completed",
+                "video_url": MOCK_VIDEO_URL
+            }
+        
         try:
             # Get operation status using the operation object
             # First, we need to recreate the operation from the name