npcpy 1.3.18__py3-none-any.whl → 1.3.19__py3-none-any.whl

npcpy/serve.py

@@ -58,7 +58,7 @@ import base64
 import shutil
 import uuid
 
-from npcpy.llm_funcs import gen_image, breathe                                                                                                                                                                
+from npcpy.llm_funcs import gen_image, gen_video, breathe                                                                                                                                                                
 
 from sqlalchemy import create_engine, text
 from sqlalchemy.orm import sessionmaker
@@ -4333,6 +4333,89 @@ def get_image_models_api():
         return jsonify({"models": [], "error": str(e)}), 500
 
 
+@app.route("/api/generate_video", methods=["POST"])
+def generate_video_api():
+    """
+    API endpoint for video generation.
+    """
+    try:
+        data = request.get_json()
+        prompt = data.get("prompt", "")
+        model = data.get("model", "veo-3.1-generate-preview")
+        provider = data.get("provider", "gemini")
+        duration = data.get("duration", 5)
+        output_dir = data.get("output_dir")  # Optional user-specified path
+        negative_prompt = data.get("negative_prompt", "")
+        reference_image = data.get("reference_image")  # Optional base64 image
+
+        if not prompt:
+            return jsonify({"error": "Prompt is required"}), 400
+
+        # Create output directory - use user-specified path or default to ~/.npcsh/videos
+        if output_dir:
+            save_dir = os.path.expanduser(output_dir)
+        else:
+            save_dir = os.path.expanduser("~/.npcsh/videos")
+        os.makedirs(save_dir, exist_ok=True)
+
+        # Generate unique filename
+        timestamp = datetime.datetime.now().strftime("%Y%m%d_%H%M%S")
+        output_filename = f"video_{timestamp}.mp4"
+        output_path = os.path.join(save_dir, output_filename)
+
+        # Calculate num_frames based on duration (assuming ~25fps for diffusers)
+        num_frames = int(duration * 25) if provider == "diffusers" else 25
+
+        print(f"Generating video with model={model}, provider={provider}, duration={duration}s")
+
+        result = gen_video(
+            prompt=prompt,
+            model=model,
+            provider=provider,
+            output_path=output_path,
+            num_frames=num_frames,
+            negative_prompt=negative_prompt,
+        )
+
+        if result and "output" in result:
+            # Read the generated video file and encode to base64
+            video_path = output_path
+            if os.path.exists(video_path):
+                with open(video_path, "rb") as f:
+                    video_data = f.read()
+                video_base64 = base64.b64encode(video_data).decode("utf-8")
+
+                return jsonify({
+                    "success": True,
+                    "video_path": video_path,
+                    "video_base64": f"data:video/mp4;base64,{video_base64}",
+                    "message": result.get("output", "Video generated successfully")
+                })
+            else:
+                return jsonify({"error": "Video file was not created"}), 500
+        else:
+            return jsonify({"error": result.get("output", "Video generation failed")}), 500
+
+    except Exception as e:
+        print(f"Error generating video: {e}")
+        traceback.print_exc()
+        return jsonify({"error": str(e)}), 500
+
+
+@app.route("/api/video_models", methods=["GET"])
+def get_video_models_api():
+    """
+    API endpoint to retrieve available video generation models.
+    """
+    video_models = [
+        # Google Veo via Gemini API (requires GEMINI_API_KEY)
+        {"value": "veo-3.1-generate-preview", "display_name": "Veo 3.1 | gemini", "provider": "gemini", "max_duration": 8},
+        {"value": "veo-3.1-fast-generate-preview", "display_name": "Veo 3.1 Fast | gemini", "provider": "gemini", "max_duration": 8},
+        {"value": "veo-2.0-generate-001", "display_name": "Veo 2 | gemini", "provider": "gemini", "max_duration": 8},
+        # Diffusers - damo-vilab/text-to-video-ms-1.7b (local)
+        {"value": "damo-vilab/text-to-video-ms-1.7b", "display_name": "ModelScope 1.7B (Local) | diffusers", "provider": "diffusers", "max_duration": 4},
+    ]
+    return jsonify({"models": video_models, "error": None})
 
 
 

{npcpy-1.3.18.dist-info → npcpy-1.3.19.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: npcpy
-Version: 1.3.18
+Version: 1.3.19
 Summary: npcpy is the premier open-source library for integrating LLMs and Agents into python systems.
 Home-page: https://github.com/NPC-Worldwide/npcpy
 Author: Christopher Agostino

{npcpy-1.3.18.dist-info → npcpy-1.3.19.dist-info}/RECORD

@@ -7,7 +7,7 @@ npcpy/npc_array.py,sha256=5qjaA9KjmJ_Zk_VxLrCyVrj73aDXpm3iJf0ngq1yIJk,45721
 npcpy/npc_compiler.py,sha256=6-SYOddpi2jTJ1KtbMxtNGC7ksaiuuLd59bmc0eiOUA,121619
 npcpy/npc_sysenv.py,sha256=1E2zwMj7aPrtRJuJSowGkvNApi07Vue3FhXsipi1XDs,45251
 npcpy/npcs.py,sha256=eExuVsbTfrRobTRRptRpDm46jCLWUgbvy4_U7IUQo-c,744
-npcpy/serve.py,sha256=3aCJe7i1eOg5w0mTLhz0JXTZpUcWnZXlmaogPGyO1Ko,275693
+npcpy/serve.py,sha256=G8slQRslfOnGnSUD8GP1IOiqQhUWu6vF_azClvI7jLs,279284
 npcpy/tools.py,sha256=A5_oVmZkzGnI3BI-NmneuxeXQq-r29PbpAZP4nV4jrc,5303
 npcpy/data/__init__.py,sha256=1tcoChR-Hjn905JDLqaW9ElRmcISCTJdE7BGXPlym2Q,642
 npcpy/data/audio.py,sha256=o4auV8DQrAmZ4y84U3SofiwEuq5-ZBjGEZipQ9zPpGQ,22816
@@ -53,8 +53,8 @@ npcpy/work/browser.py,sha256=p2PeaoZdAXipFuAgKCCB3aXXLE_p3yIRqC87KlZKZWc,679
 npcpy/work/desktop.py,sha256=F3I8mUtJp6LAkXodsh8hGZIncoads6c_2Utty-0EdDA,2986
 npcpy/work/plan.py,sha256=QyUwg8vElWiHuoS-xK4jXTxxHvkMD3VkaCEsCmrEPQk,8300
 npcpy/work/trigger.py,sha256=P1Y8u1wQRsS2WACims_2IdkBEar-iBQix-2TDWoW0OM,9948
-npcpy-1.3.18.dist-info/licenses/LICENSE,sha256=j0YPvce7Ng9e32zYOu0EmXjXeJ0Nwawd0RA3uSGGH4E,1070
-npcpy-1.3.18.dist-info/METADATA,sha256=FBBa4cRcrD0tfenjd5nGxtxk-P8bRc-dRlR6s8PSXkM,37870
-npcpy-1.3.18.dist-info/WHEEL,sha256=qELbo2s1Yzl39ZmrAibXA2jjPLUYfnVhUNTlyF1rq0Y,92
-npcpy-1.3.18.dist-info/top_level.txt,sha256=g1pbSvrOOncB74Bg5-J0Olg4V0A5VzDw-Xz5YObq8BU,6
-npcpy-1.3.18.dist-info/RECORD,,
+npcpy-1.3.19.dist-info/licenses/LICENSE,sha256=j0YPvce7Ng9e32zYOu0EmXjXeJ0Nwawd0RA3uSGGH4E,1070
+npcpy-1.3.19.dist-info/METADATA,sha256=18bGYxznkKrwP59mZ_06rQfMFQBe_hdAjB_VbHyZBYY,37870
+npcpy-1.3.19.dist-info/WHEEL,sha256=qELbo2s1Yzl39ZmrAibXA2jjPLUYfnVhUNTlyF1rq0Y,92
+npcpy-1.3.19.dist-info/top_level.txt,sha256=g1pbSvrOOncB74Bg5-J0Olg4V0A5VzDw-Xz5YObq8BU,6
+npcpy-1.3.19.dist-info/RECORD,,