Update app.py

app.py

@@ -159,16 +159,24 @@ def generate_video(
                 enhance_prompt=enhance_prompt,
                 input_image=image_input,
                 seed=current_seed,
+                negative_prompt=negative_prompt,
                 api_name="/encode_prompt"
             )
             embedding_path = result[0]  # Path to .pt file
             print(f"Embeddings received from: {embedding_path}")
-    
+
             # Load embeddings
             embeddings = torch.load(embedding_path)
-            video_context = embeddings['video_context']
-            audio_context = embeddings['audio_context']
+            video_context_positive = embeddings['video_context']
+            audio_context_positive = embeddings['audio_context']
+
+            # Load negative contexts if available
+            video_context_negative = embeddings.get('video_context_negative', None)
+            audio_context_negative = embeddings.get('audio_context_negative', None)
+
             print("✓ Embeddings loaded successfully")
+            if video_context_negative is not None:
+                print("  ✓ Negative prompt embeddings also loaded")
         except Exception as e:
             raise RuntimeError(
                 f"Failed to get embeddings from text encoder space: {e}\n"
@@ -189,8 +197,10 @@ def generate_video(
             cfg_guidance_scale=cfg_guidance_scale,
             images=images,
             tiling_config=TilingConfig.default(),
-            video_context=video_context,
-            audio_context=audio_context,
+            video_context_positive=video_context_positive,
+            audio_context_positive=audio_context_positive,
+            video_context_negative=video_context_negative,
+            audio_context_negative=audio_context_negative,
         )
 
         return str(output_path), current_seed