fix: replace AutoPipelineForText2Video with DiffusionPipeline and add custom AI Mind Settings to prompt composer

app.py

@@ -63,18 +63,24 @@ def get_device_info():
 
 DEVICE = get_device_info()
 
-def enhance_prompt_with_ai(prompt_text, style_preset):
+def enhance_prompt_with_ai(prompt_text, style_preset, ai_url, ai_key, session_id):
     if not prompt_text or not prompt_text.strip():
         return "Please enter a prompt first."
         
-    url = "https://invictatill-invictatill-ai.hf.space/api/v1/chat"
-    
     style_modifiers = STYLE_PRESETS.get(style_preset, "")
     
+    # Standardize URL
+    ai_url = (ai_url or "").strip().rstrip("/")
+    if not ai_url:
+        ai_url = "https://invictatill-invictatill-ai.hf.space"
+    
+    chat_endpoint = f"{ai_url}/api/v1/chat"
+    
     system_prompt = (
         "You are an expert cinematic prompt engineer for video generation models (like Wan 2.1, LTX-Video, CogVideo). "
         "Your task is to rewrite the user's simple prompt into a highly descriptive, visually stunning, "
         "and detailed prompt optimized for text-to-video models. "
+        "Use your learned facts, memory context, and knowledge about the user's business if relevant. "
         "Include specific details about lighting, camera angle, motion, and atmosphere. "
         "Keep it under 75 words. "
         "Respond ONLY with the final enhanced prompt. Do NOT include any intro or conversational filler."
@@ -87,9 +93,15 @@ def enhance_prompt_with_ai(prompt_text, style_preset):
     payload = {
         "message": f"[SYSTEM CONTEXT]\n{system_prompt}\n\nUser: {full_prompt}\n\nAssistant:"
     }
-    
+    if session_id:
+        payload["session_id"] = session_id
+        
+    headers = {"Content-Type": "application/json"}
+    if ai_key:
+        headers["Authorization"] = f"Bearer {ai_key}"
+            
     try:
-        response = requests.post(url, json=payload, timeout=8)
+        response = requests.post(chat_endpoint, json=payload, headers=headers, timeout=12)
         if response.status_code == 200:
             data = response.json()
             enhanced = data.get("reply") or data.get("choices", [{}])[0].get("message", {}).get("content", "")
@@ -116,26 +128,27 @@ def load_model(model_name: str, progress=gr.Progress()):
     progress(0.2, desc=f"Downloading/Loading {model_name}...")
     
     try:
+        from diffusers import DiffusionPipeline
         if "wan" in model_id.lower():
             try:
                 from diffusers import WanPipeline
                 pipe = WanPipeline.from_pretrained(model_id, torch_dtype=DEVICE["dtype"])
             except ImportError:
-                from diffusers import AutoPipelineForText2Video
-                pipe = AutoPipelineForText2Video.from_pretrained(model_id, torch_dtype=DEVICE["dtype"])
+                pipe = DiffusionPipeline.from_pretrained(model_id, torch_dtype=DEVICE["dtype"])
         elif "ltx" in model_id.lower():
             try:
                 from diffusers import LTXVideoPipeline
                 pipe = LTXVideoPipeline.from_pretrained(model_id, torch_dtype=DEVICE["dtype"])
             except ImportError:
-                from diffusers import AutoPipelineForText2Video
-                pipe = AutoPipelineForText2Video.from_pretrained(model_id, torch_dtype=DEVICE["dtype"])
+                pipe = DiffusionPipeline.from_pretrained(model_id, torch_dtype=DEVICE["dtype"])
         elif "cogvideo" in model_id.lower():
-            from diffusers import CogVideoXPipeline
-            pipe = CogVideoXPipeline.from_pretrained(model_id, torch_dtype=DEVICE["dtype"])
+            try:
+                from diffusers import CogVideoXPipeline
+                pipe = CogVideoXPipeline.from_pretrained(model_id, torch_dtype=DEVICE["dtype"])
+            except ImportError:
+                pipe = DiffusionPipeline.from_pretrained(model_id, torch_dtype=DEVICE["dtype"])
         else:
-            from diffusers import AutoPipelineForText2Video
-            pipe = AutoPipelineForText2Video.from_pretrained(
+            pipe = DiffusionPipeline.from_pretrained(
                 model_id,
                 torch_dtype=DEVICE["dtype"],
                 variant="fp16" if DEVICE["device"] == "cuda" else None,
@@ -387,8 +400,27 @@ with gr.Blocks(css=custom_css, title="InvictaTill VideoGen Studio", theme=gr.the
             gr.Markdown("### ✍️ Prompt Composer")
             prompt = gr.Textbox(label="Describe your scene", placeholder="A cyberpunk drone shot flying through neon-lit Tokyo streets...", lines=4, elem_id="prompt")
             
+            with gr.Accordion("🧠 InvictaTill AI Mind Integration Settings", open=False):
+                gr.Markdown("Configure the endpoint URL and API Key for your running InvictaTill AI Space instance so the prompt enhancer can read your business insights and custom memories.")
+                ai_url_input = gr.Textbox(
+                    value=os.environ.get("VITE_INVICTATILL_AI_URL", "https://invictatill-invictatill-ai.hf.space"),
+                    label="AI Mind Endpoint URL",
+                    placeholder="https://invictatill-invictatill-ai.hf.space"
+                )
+                ai_key_input = gr.Textbox(
+                    value=os.environ.get("VITE_INVICTATILL_AI_KEY", ""),
+                    label="API Key / Auth Token",
+                    placeholder="invicta_sk_...",
+                    type="password"
+                )
+                session_id_input = gr.Textbox(
+                    value="videogen_studio_session",
+                    label="Session ID (Loads Memory Context)",
+                    placeholder="videogen_studio_session"
+                )
+            
             with gr.Row():
-                enhance_btn = gr.Button("✨ Enhance Prompt with InvictaTill AI", variant="secondary")
+                enhance_btn = gr.Button("✨ Enhance Prompt with InvictaTill AI Mind", variant="secondary")
                 
             negative_prompt = gr.Textbox(label="Negative Prompt", placeholder="blur, distortion, low quality, watermark", lines=2, value="blur, distortion, low quality, watermark, text, bad anatomy, deformed, cartoonish, static")
             
@@ -401,7 +433,7 @@ with gr.Blocks(css=custom_css, title="InvictaTill VideoGen Studio", theme=gr.the
     # Click Handlers
     enhance_btn.click(
         fn=enhance_prompt_with_ai,
-        inputs=[prompt, style_dropdown],
+        inputs=[prompt, style_dropdown, ai_url_input, ai_key_input, session_id_input],
         outputs=[prompt]
     )