Update video_generation.py

video_generation.py

@@ -18,32 +18,52 @@ class FreeVideoGenerator:
         self.hf_token = hf_token or os.getenv('HF_TOKEN', '')
         
         # Available free models
-        self.text_to_video_models = [
+        self.models = [
             "cerspense/zeroscope_v2_576w",
             "damo-vilab/text-to-video-ms-1.7b"
         ]
         
-        self.current_model = self.text_to_video_models[0]
+        self.current_model = self.models[0]
         self.timeout = 120
         self.max_retries = 2
         
+        # Video settings
+        self.width = 576
+        self.height = 320
+        self.fps = 8
+        self.frames = 24
+        
     def enhance_prompt_with_context(self, prompt: str) -> str:
         """Enhance video prompts with cinematic context"""
-        cinematic_enhancements = [
+        
+        # Cinematic enhancements
+        cinematic = [
             "cinematic, 8k, ultra detailed, high quality, masterpiece",
             "epic, dramatic lighting, film grain, cinematic shot, professional",
             "beautiful, stunning, visually striking, vivid colors, trending",
             "high resolution, detailed, sharp focus, studio quality"
         ]
         
-        enhanced = prompt
-        enhanced += f", {random.choice(cinematic_enhancements)}, 576x320 resolution, 8 fps"
+        # Cultural enhancements for African themes
+        if any(word in prompt.lower() for word in ['africa', 'kenya', 'tanzania', 'safari', 'wildlife', 'cultural']):
+            cultural = [
+                "African style, vibrant colors, cultural elements, traditional",
+                "East African landscape, warm colors, cultural symbolism",
+                "African documentary style, natural lighting, authentic",
+                "Traditional African art style, symbolic, meaningful"
+            ]
+            enhanced = f"{prompt}, {random.choice(cinematic)}, {random.choice(cultural)}"
+        else:
+            enhanced = f"{prompt}, {random.choice(cinematic)}"
+        
+        # Add technical specs
+        enhanced += f", {self.width}x{self.height} resolution, {self.fps} fps"
         
         return enhanced
     
     def generate_text_to_video(self, prompt: str) -> Optional[str]:
         """
-        Generate video from text prompt
+        Generate video from text prompt using Hugging Face API
         """
         try:
             enhanced_prompt = self.enhance_prompt_with_context(prompt)
@@ -55,12 +75,13 @@ class FreeVideoGenerator:
             payload = {
                 "inputs": enhanced_prompt,
                 "parameters": {
-                    "num_frames": 24,
+                    "num_frames": self.frames,
                     "num_inference_steps": 25,
                     "guidance_scale": 7.5,
-                    "fps": 8,
-                    "height": 320,
-                    "width": 576
+                    "fps": self.fps,
+                    "height": self.height,
+                    "width": self.width,
+                    "negative_prompt": "blurry, low quality, distorted, watermark, text"
                 }
             }
             
@@ -91,44 +112,87 @@ class FreeVideoGenerator:
                 except requests.exceptions.Timeout:
                     logger.warning(f"⏰ Request timeout, attempt {attempt + 1}")
                     continue
+                except Exception as e:
+                    logger.error(f"Video generation error: {e}")
+                    break
                     
         except Exception as e:
-            logger.error(f"Video generation error: {e}")
+            logger.error(f"Video generation failed: {e}")
         
         return None
     
     def create_cultural_video(self, theme: str, style: str = "animated") -> Optional[str]:
         """
-        Create videos with cultural themes
+        Create videos with African cultural themes
         """
+        # Cultural themes database
         cultural_themes = {
-            "safari": "African safari sunset with elephants and giraffes, majestic savanna landscape",
-            "dance": "Traditional Maasai warriors dancing, vibrant colors, cultural celebration",
-            "market": "Busy African market scene, vibrant colors, people trading goods",
-            "coastal": "Swahili coast with traditional dhows sailing, Indian Ocean waves",
-            "wildlife": "African wildlife documentary style, lions hunting on savanna",
-            "village": "Traditional African village life, community activities, sunset"
+            "safari": {
+                "prompt": "African safari sunset with elephants and giraffes walking, majestic savanna landscape, acacia trees, warm colors",
+                "styles": {
+                    "animated": "animated, cartoon style, smooth motion, vibrant colors",
+                    "realistic": "realistic, documentary style, cinematic, natural lighting"
+                }
+            },
+            "dance": {
+                "prompt": "Traditional Maasai warriors dancing, vibrant colors, cultural celebration, energetic movement, community",
+                "styles": {
+                    "animated": "animated, lively motion, colorful, celebratory",
+                    "realistic": "realistic, documentary footage, authentic, cultural"
+                }
+            },
+            "market": {
+                "prompt": "Busy African market scene, vibrant colors, people trading goods, lively atmosphere, traditional clothing",
+                "styles": {
+                    "animated": "animated, bustling market, colorful stalls, lively",
+                    "realistic": "realistic, documentary style, authentic market scene"
+                }
+            },
+            "coastal": {
+                "prompt": "Swahili coast with traditional dhows sailing, Indian Ocean waves, beach scenery, palm trees, traditional architecture",
+                "styles": {
+                    "animated": "animated, ocean waves, sailing dhows, coastal life",
+                    "realistic": "realistic, coastal documentary, ocean scenery"
+                }
+            },
+            "wildlife": {
+                "prompt": "African wildlife documentary style, lions hunting on savanna, dramatic nature scene, wildlife behavior",
+                "styles": {
+                    "animated": "animated, wildlife cartoon, animal movement",
+                    "realistic": "realistic, nature documentary, wildlife footage"
+                }
+            },
+            "village": {
+                "prompt": "Traditional African village life, community activities, sunset over huts, daily life, cultural activities",
+                "styles": {
+                    "animated": "animated, village life, community activities",
+                    "realistic": "realistic, documentary style, authentic village"
+                }
+            }
         }
         
-        base_prompt = cultural_themes.get(theme, f"African {theme}, cultural, vibrant")
-        
-        style_enhancements = {
-            "animated": "animated, cartoon style, smooth motion, vibrant colors",
-            "realistic": "realistic, documentary style, cinematic, natural lighting"
-        }
+        # Get theme data
+        theme_data = cultural_themes.get(theme, cultural_themes["safari"])
+        base_prompt = theme_data["prompt"]
+        style_enhancement = theme_data["styles"].get(style, "animated, vibrant colors")
         
-        full_prompt = f"{base_prompt}, {style_enhancements.get(style, 'animated, vibrant')}"
+        # Combine prompt
+        full_prompt = f"{base_prompt}, {style_enhancement}, cultural, authentic, East African"
         
         return self.generate_text_to_video(full_prompt)
     
     def get_video_info(self) -> dict:
-        """Get information about video generation"""
+        """Get information about video generation capabilities"""
         return {
-            "available_models": self.text_to_video_models,
+            "available_models": self.models,
             "current_model": self.current_model,
-            "resolution": "576x320",
-            "fps": 8,
-            "max_frames": 24,
-            "max_duration": "3 seconds",
-            "free": True
+            "resolution": f"{self.width}x{self.height}",
+            "fps": self.fps,
+            "max_frames": self.frames,
+            "max_duration": f"{self.frames/self.fps:.1f} seconds",
+            "cultural_themes": ["safari", "dance", "market", "coastal", "wildlife", "village"],
+            "styles": ["animated", "realistic"],
+            "free": True,
+            "provider": "Hugging Face Inference API",
+            "creator": "Stanley Samwel Owino"
         }