Refactor code, remove deepseek integration

src/api_clients.py

@@ -82,29 +82,29 @@ class APIClients:
             Local path to generated image or None
         """
         try:
-            # import vertexai
-            # from vertexai.preview.vision_models import ImageGenerationModel
+            import vertexai
+            from vertexai.preview.vision_models import ImageGenerationModel
 
-            # logger.info(f"🎨 Generating image with Imagen 4 Ultra: {prompt[:200]}...")
+            logger.info(f"🎨 Generating image with Imagen 4 Ultra: {prompt[:200]}...")
 
-            # vertexai.init(project=self.config.get("gcp_project_id"), location="us-central1")
+            vertexai.init(project=self.config.get("gcp_project_id"), location="us-central1")
 
-            # # Use correct Imagen 4 Ultra model name
-            # model = ImageGenerationModel.from_pretrained("imagen-4.0-ultra-generate-001")
+            # Use correct Imagen 4 Ultra model name
+            model = ImageGenerationModel.from_pretrained("imagen-4.0-ultra-generate-001")
 
-            # images = model.generate_images(
-            #     prompt=prompt,
-            #     number_of_images=1,
-            #     aspect_ratio="9:16",  # Vertical for TikTok/Instagram
-            #     safety_filter_level="block_some",
-            #     person_generation="allow_adult",
-            # )
+            images = model.generate_images(
+                prompt=prompt,
+                number_of_images=1,
+                aspect_ratio="9:16",  # Vertical for TikTok/Instagram
+                safety_filter_level="block_some",
+                person_generation="allow_adult",
+            )
 
-            # # Save to temp file
-            # import tempfile
-            # output_path = f"/tmp/hook_image_{hash(prompt)}.png"
-            # images[0].save(location=output_path, include_generation_parameters=False)
-            output_path = '/Users/topcoderkz/Downloads/gen4-ultra.png'
+            # Save to temp file
+            import tempfile
+            output_path = f"/tmp/hook_image_{hash(prompt)}.png"
+            images[0].save(location=output_path, include_generation_parameters=False)
+            # output_path = '/Users/topcoderkz/Downloads/gen4-ultra.png'
             logger.info(f"✓ Image generated with Imagen 4 Ultra: {output_path}")
             return output_path
 
@@ -190,76 +190,76 @@ class APIClients:
         """
         try:
             logger.info(f"🎬 Generating video with gen4_turbo: {prompt[:100]}...")
-            return {
-                "video_url": 'https://dnznrvs05pmza.cloudfront.net/764d8b31-4e1f-4ba2-bf4f-360cf029e0b7.mp4?_jwt=eyJhbGciOiJIUzI1NiIsInR5cCI6IkpXVCJ9.eyJrZXlIYXNoIjoiMGIyZjMyMzc5NDA4ZTU0NCIsImJ1Y2tldCI6InJ1bndheS10YXNrLWFydGlmYWN0cyIsInN0YWdlIjoicHJvZCIsImV4cCI6MTc2MDQwMDAwMH0.5ltZPfO-gWilm_lt6sK-tPvgwJUgPluOjeeUOOIbyEE',
-                "task_id": '08fbc334-7d36-45c2-8b71-7f20fa075f10',
-                "duration": duration,
-                "prompt": prompt,
-                "status": 'SUCCEEDED',
-                "created_at": '2025-10-12T18:57:27.240Z',
-                "model": "gen4_turbo"
-            }
-
-            # headers = {
-            #     "Authorization": f"Bearer {self.runway_api_key}",
-            #     "Content-Type": "application/json",
-            #     "X-Runway-Version": "2024-11-06",
-            # }
-
-            # payload = {
-            #     "promptImage": image_url,
-            #     "promptText": prompt[:1000],
-            #     "model": "gen4_turbo",  # Updated to gen4_turbo ($0.25/video)
+            # return {
+            #     "video_url": 'https://dnznrvs05pmza.cloudfront.net/764d8b31-4e1f-4ba2-bf4f-360cf029e0b7.mp4?_jwt=eyJhbGciOiJIUzI1NiIsInR5cCI6IkpXVCJ9.eyJrZXlIYXNoIjoiMGIyZjMyMzc5NDA4ZTU0NCIsImJ1Y2tldCI6InJ1bndheS10YXNrLWFydGlmYWN0cyIsInN0YWdlIjoicHJvZCIsImV4cCI6MTc2MDQwMDAwMH0.5ltZPfO-gWilm_lt6sK-tPvgwJUgPluOjeeUOOIbyEE',
+            #     "task_id": '08fbc334-7d36-45c2-8b71-7f20fa075f10',
             #     "duration": duration,
-            #     "ratio": "1280:720"
+            #     "prompt": prompt,
+            #     "status": 'SUCCEEDED',
+            #     "created_at": '2025-10-12T18:57:27.240Z',
+            #     "model": "gen4_turbo"
             # }
 
-            # async with aiohttp.ClientSession() as session:
-            #     # Create task
-            #     async with session.post(
-            #         "https://api.dev.runwayml.com/v1/image_to_video", 
-            #         headers=headers, 
-            #         json=payload
-            #     ) as response:
-            #         if response.status != 200:
-            #             error_text = await response.text()
-            #             raise Exception(f"RunwayML error: {error_text}")
-
-            #         task_data = await response.json()
-            #         task_id = task_data["id"]
-            #         logger.info(f"✓ Task created with gen4_turbo: {task_id}")
-
-            #     # Poll for completion
-            #     max_attempts = 120
-            #     for attempt in range(max_attempts):
-            #         await asyncio.sleep(10)
-
-            #         async with session.get(
-            #             f"https://api.dev.runwayml.com/v1/tasks/{task_id}", 
-            #             headers=headers
-            #         ) as status_response:
-            #             status_data = await status_response.json()
-            #             status = status_data["status"]
-
-            #             if status == "SUCCEEDED":
-            #                 video_url = status_data["output"][0]
-            #                 logger.info(f"✅ Video generated with gen4_turbo: {video_url}")
-            #                 return {
-            #                     "video_url": video_url,
-            #                     "task_id": task_id,
-            #                     "duration": duration,
-            #                     "prompt": prompt,
-            #                     "status": status,
-            #                     "created_at": status_data.get("createdAt"),
-            #                     "model": "gen4_turbo"
-            #                 }
-            #             elif status == "FAILED":
-            #                 raise Exception(f"Generation failed: {status_data.get('failure')}")
-            #             elif status == "RUNNING":
-            #                 progress = status_data.get("progress", 0)
-            #                 logger.info(f"⏳ Progress: {progress*100:.0f}%")
-
-            #     raise Exception("Timeout waiting for video generation")
+            headers = {
+                "Authorization": f"Bearer {self.runway_api_key}",
+                "Content-Type": "application/json",
+                "X-Runway-Version": "2024-11-06",
+            }
+
+            payload = {
+                "promptImage": image_url,
+                "promptText": prompt[:1000],
+                "model": "gen4_turbo",  # Updated to gen4_turbo ($0.25/video)
+                "duration": duration,
+                "ratio": "1280:720"
+            }
+
+            async with aiohttp.ClientSession() as session:
+                # Create task
+                async with session.post(
+                    "https://api.dev.runwayml.com/v1/image_to_video", 
+                    headers=headers, 
+                    json=payload
+                ) as response:
+                    if response.status != 200:
+                        error_text = await response.text()
+                        raise Exception(f"RunwayML error: {error_text}")
+
+                    task_data = await response.json()
+                    task_id = task_data["id"]
+                    logger.info(f"✓ Task created with gen4_turbo: {task_id}")
+
+                # Poll for completion
+                max_attempts = 120
+                for attempt in range(max_attempts):
+                    await asyncio.sleep(10)
+
+                    async with session.get(
+                        f"https://api.dev.runwayml.com/v1/tasks/{task_id}", 
+                        headers=headers
+                    ) as status_response:
+                        status_data = await status_response.json()
+                        status = status_data["status"]
+
+                        if status == "SUCCEEDED":
+                            video_url = status_data["output"][0]
+                            logger.info(f"✅ Video generated with gen4_turbo: {video_url}")
+                            return {
+                                "video_url": video_url,
+                                "task_id": task_id,
+                                "duration": duration,
+                                "prompt": prompt,
+                                "status": status,
+                                "created_at": status_data.get("createdAt"),
+                                "model": "gen4_turbo"
+                            }
+                        elif status == "FAILED":
+                            raise Exception(f"Generation failed: {status_data.get('failure')}")
+                        elif status == "RUNNING":
+                            progress = status_data.get("progress", 0)
+                            logger.info(f"⏳ Progress: {progress*100:.0f}%")
+
+                raise Exception("Timeout waiting for video generation")
 
         except Exception as e:
             logger.error(f"❌ Video generation error: {e}")