media-agent-mcp 2.7.0__py3-none-any.whl → 2.7.2__py3-none-any.whl

media_agent_mcp/ai_models/omni_human.py

@@ -122,8 +122,8 @@ if __name__ == '__main__':
     time1 = time.time()
 
     print(generate_video_from_omni_human(
-        image_url="https://ark-content-generation-v2-ap-southeast-1.tos-ap-southeast-1.volces.com/seedream-3-0-t2i/0217564979274996d76cd7acfd572da46abbcedd46e0f23a31db6.jpeg?X-Tos-Algorithm=TOS4-HMAC-SHA256&X-Tos-Credential=AKLTYWJkZTExNjA1ZDUyNDc3YzhjNTM5OGIyNjBhNDcyOTQ%2F20250829%2Fap-southeast-1%2Ftos%2Frequest&X-Tos-Date=20250829T200528Z&X-Tos-Expires=86400&X-Tos-Signature=abf8c7d516eecd4543691ffee5c3c5bcc7e38d5d047caf7f2fcfe3e41ab897d3&X-Tos-SignedHeaders=host",
-        audio_url="https://carey.tos-ap-southeast-1.bytepluses.com/media_agent/2025-08-30/a72d04349e81454c8065674792256a1d.mp3"
+        image_url="https://carey.tos-ap-southeast-1.bytepluses.com/Art%20Portrait/Art%20Portrait/Art%20Portrait/Art%20Portrait%20(1).jpg",
+        audio_url="https://carey.tos-ap-southeast-1.bytepluses.com/media_agent/2025-09-02/66620bed2a5f4b2cbc641559ff93a2ed.mp3"
     ))
 
     print(time.time() - time1)

media_agent_mcp/async_server.py

@@ -449,8 +449,6 @@ def main():
                         help='Host for SSE transport (default: 127.0.0.1)')
     parser.add_argument('--port', type=int, default=8000,
                         help='Port for SSE transport (default: 8000)')
-    parser.add_argument('--version', action='store_true',
-                        help='Show version information')
     parser.add_argument('--run-be', action='store_true',
                         help='Run the backend server')
     parser.add_argument('--be-host', type=str, default='0.0.0.0',
@@ -460,10 +458,6 @@ def main():
 
     args = parser.parse_args()
 
-    if args.version:
-        print("Async Media Agent MCP Server v0.1.0")
-        return
-
     if args.run_be:
         logger.info(f"Starting backend server on {args.be_host}:{args.be_port}...")
         be_path = os.path.abspath(os.path.join(os.path.dirname(__file__), 'be'))
@@ -473,22 +467,20 @@ def main():
         try:
             subprocess.run(
                 [
-                    sys.executable,
-                    "-c",
-                    (
-                        "import sys, app; "
-                        "flask_app = getattr(app, '_flask_app', getattr(app, 'app', None)); "
-                        "assert flask_app is not None, 'No Flask application instance found in app.py'; "
-                        "flask_app.run(host=sys.argv[1], port=int(sys.argv[2]), debug=False)"
-                    ),
-                    args.be_host,
-                    str(args.be_port),
+                    "gunicorn",
+                    "--workers",
+                    "4",
+                    "--bind",
+                    f"{args.be_host}:{args.be_port}",
+                    "--log-level",
+                    "debug",
+                    "app:app",
                 ],
                 cwd=be_path,
                 check=True
             )
         except FileNotFoundError:
-            logger.error("`uvicorn` command not found. Please ensure it is installed in your environment.")
+            logger.error("`gunicorn` command not found. Please ensure it is installed in your environment.")
         except subprocess.CalledProcessError as e:
             logger.error(f"Failed to start backend server: {e}")
         return
@@ -528,7 +520,7 @@ def main():
                 mcp.sse_app(),
                 host=args.host,
                 port=args.port,
-                timeout_keep_alive=1200
+                timeout_keep_alive=300
             )
         else:
             # Default stdio transport

media_agent_mcp/be/app.py

@@ -6,6 +6,7 @@ from flask import Flask
 # Register blueprints from modularized routes
 from media_agent_mcp.be.routes_media import media_bp
 from media_agent_mcp.be.routes_subtitles import subtitles_bp
+from media_agent_mcp.be.routes_omni_human import omni_human_bp
 
 
 logger = logging.getLogger(__name__)
@@ -14,6 +15,7 @@ app = Flask(__name__)
 # Register blueprints (keep original paths unchanged)
 app.register_blueprint(media_bp)
 app.register_blueprint(subtitles_bp)
+app.register_blueprint(omni_human_bp)
 
 
 if __name__ == "__main__":

media_agent_mcp/be/routes_media.py

@@ -5,6 +5,7 @@ import subprocess
 import tempfile
 from pathlib import Path
 from typing import List, Optional, Dict, Any
+from concurrent.futures import ThreadPoolExecutor
 
 from flask import Blueprint, jsonify, request, send_file, after_this_request
 
@@ -160,13 +161,15 @@ def concat_videos():
                     pass
             return response
 
-        logger.info('Downloading videos')
+        logger.info('Downloading videos concurrently')
         video_paths = []
-        for url in video_urls:
-            dl = download_video_from_url(url)
+        with ThreadPoolExecutor(max_workers=min(len(video_urls), 5)) as executor:
+            download_results = list(executor.map(download_video_from_url, video_urls))
+
+        for dl in download_results:
             if dl.get("status") == "error":
                 return jsonify(dl), 400
-            path = Path(dl["data"]["file_path"])  # type: ignore[index]
+            path = Path(dl["data"]["file_path"])
             temp_files.append(path)
             video_paths.append(path)
 

media_agent_mcp/be/routes_omni_human.py

@@ -0,0 +1,34 @@
+from flask import Blueprint, request, jsonify
+from media_agent_mcp.video.omni_human import generate_video_from_omni_human
+
+omni_human_bp = Blueprint("omni_human", __name__)
+
+@omni_human_bp.post("/generate-video-from-omni-human")
+def generate_video():
+    """
+    Generates a video from an image and audio using the Omni Human API.
+    """
+    try:
+        data = request.get_json(silent=True) or {}
+        image_url = data.get("image_url")
+        audio_url = data.get("audio_url")
+
+        if not image_url or not audio_url:
+            return jsonify({
+                "status": "error",
+                "data": None,
+                "message": "Fields image_url and audio_url are required"
+            }), 400
+
+        video_url = generate_video_from_omni_human(image_url, audio_url)
+
+        return jsonify({
+            "status": "success",
+            "data": {
+                "video_url": video_url
+            },
+            "message": "Video generated successfully"
+        })
+
+    except Exception as e:
+        return jsonify({"status": "error", "data": None, "message": str(e)}), 500

media_agent_mcp/video/omni_human.py

@@ -0,0 +1,118 @@
+import hashlib
+import json
+import os
+import random
+import time
+from typing import Dict, Any
+import requests
+
+
+def _generate_signature(nonce: int, timestamp: int, security_key: str) -> str:
+    """
+    Generates a signature for the API request.
+    """
+    keys = [str(nonce), str(security_key), str(timestamp)]
+    keys.sort()
+    key_str = "".join(keys).encode("utf-8")
+    signature = hashlib.sha1(key_str).hexdigest()
+    return signature.lower()
+
+
+def _submit_task(image_url: str, audio_url: str, api_key: str, security_key: str) -> str:
+    """
+    Submits a video generation task.
+    """
+    submit_task_url = "https://cv-api.byteintlapi.com/api/common/v2/submit_task"
+    timestamp = int(time.time())
+    nonce = random.randint(0, (1 << 31) - 1)
+    signature = _generate_signature(nonce, timestamp, security_key)
+
+    params = {
+        "api_key": api_key,
+        "timestamp": str(timestamp),
+        "nonce": str(nonce),
+        "sign": signature,
+    }
+    headers = {"Content-Type": "application/json"}
+    body = {
+        "req_key": "realman_avatar_picture_omni_cv",
+        "image_url": image_url,
+        "audio_url": audio_url,
+    }
+
+    response = requests.post(submit_task_url, params=params, headers=headers, json=body)
+    response.raise_for_status()
+    data = response.json()
+    if data["code"] != 10000:
+        raise Exception(f"Failed to submit task: {data['message']}")
+    return data["data"]["task_id"]
+
+
+def _get_task_result(task_id: str, api_key: str, security_key: str) -> Dict[str, Any]:
+    """
+    Gets the result of a video generation task.
+    """
+    get_result_url = "https://cv-api.byteintlapi.com/api/common/v2/get_result"
+    timestamp = int(time.time())
+    nonce = random.randint(0, (1 << 31) - 1)
+    signature = _generate_signature(nonce, timestamp, security_key)
+
+    params = {
+        "api_key": api_key,
+        "timestamp": str(timestamp),
+        "nonce": str(nonce),
+        "sign": signature,
+    }
+    headers = {"Content-Type": "application/json"}
+    body = {
+        "req_key": "realman_avatar_picture_omni_cv",
+        "task_id": task_id,
+    }
+
+    response = requests.post(get_result_url, params=params, headers=headers, json=body)
+    print(response.text)
+    response.raise_for_status()
+    return response.json()
+
+
+def generate_video_from_omni_human(image_url: str, audio_url: str) -> str:
+    """
+    Generates a video from an image and audio using the Omni Human API.
+
+    Args:
+        image_url: The URL of the portrait image.
+        audio_url: The URL of the audio.
+
+    Returns:
+        The URL of the generated video.
+    """
+    api_key = os.environ.get("OMNI_HUMAN_AK")
+    security_key = os.environ.get("OMNI_HUMAN_SK")
+
+    if not api_key or not security_key:
+        raise ValueError("OMNI_HUMAN_AK and OMNI_HUMAN_SK environment variables must be set")
+
+    task_id = _submit_task(image_url, audio_url, api_key, security_key)
+    print('Submitted task, task_id:', task_id)
+    while True:
+        result = _get_task_result(task_id, api_key, security_key)
+        if result["code"] != 10000:
+            raise Exception(f"Failed to get task result: {result['message']}")
+
+        status = result.get("data", {}).get("status")
+        if status == "done":
+            # Parse resp_data JSON string to get video_url
+            resp_data_str = result["data"].get("resp_data", "{}")
+            try:
+                resp_data = json.loads(resp_data_str)
+                video_url = resp_data.get("video_url")
+                if video_url:
+                    return video_url
+                else:
+                    raise Exception(f"No video_url found in response: {resp_data}")
+            except json.JSONDecodeError as e:
+                raise Exception(f"Failed to parse resp_data JSON: {e}")
+        elif status in ["failed", "error"]:
+            raise Exception(f"Video generation failed: {result}")
+
+        time.sleep(5)

{media_agent_mcp-2.7.0.dist-info → media_agent_mcp-2.7.2.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: media-agent-mcp
-Version: 2.7.0
+Version: 2.7.2
 Summary: A Model Context Protocol server for media processing with AI tools
 Author-email: Media Agent Team <team@mediaagent.com>
 Keywords: mcp,ai,media,video,image,processing
@@ -26,6 +26,7 @@ Requires-Dist: loguru>=0.7.3
 Requires-Dist: imageio-ffmpeg>=0.4.0
 Requires-Dist: Flask>=3.0.0
 Requires-Dist: pydub>=0.25.1
+Requires-Dist: gunicorn>=22.0.0
 Requires-Dist: audioop-lts; python_version >= "3.13"
 Requires-Dist: google-genai>=1.33.0
 Requires-Dist: mcp==1.11.0

{media_agent_mcp-2.7.0.dist-info → media_agent_mcp-2.7.2.dist-info}/RECORD

@@ -1,8 +1,8 @@
 media_agent_mcp/__init__.py,sha256=s1rx7OkUul6PrxbxziYAV6c0AuamPlnelltQieHqX_I,340
-media_agent_mcp/async_server.py,sha256=boXtiFLw2mnr6ZWNZTRhchd4KAZXhbGRnOFqX-ZaJNk,19872
+media_agent_mcp/async_server.py,sha256=gsc_VbZd8PwzuPc9UmtTgAYPyBzh3PNoTUNM3yTsgp4,19418
 media_agent_mcp/async_wrapper.py,sha256=hiiBhhz9WeVDfSBWVh6ovhf5jeP5ZbsieBbz9P-KPn0,15351
 media_agent_mcp/ai_models/__init__.py,sha256=2kHzTYwjQw89U4QGDq0e2WqJScqDkDNlDaWHGak5JeY,553
-media_agent_mcp/ai_models/omni_human.py,sha256=d6fll3qqj41Wmv6K0dd8iA_LL7kJWqHuEgxbmqf9JbY,4670
+media_agent_mcp/ai_models/omni_human.py,sha256=WAWO6pI4go9FLpVwxMvizz1Ays66CMUi8XkGZmIu08w,4337
 media_agent_mcp/ai_models/openaiedit.py,sha256=uu4d2BgXSrjWRdNPs_SryI9muxO93pItVtEze9nDhjc,9776
 media_agent_mcp/ai_models/seed16.py,sha256=cX0ZONj2Jpu_dzSIq8oXSJfnsfGWVcaEmWyRxg6jMfQ,5110
 media_agent_mcp/ai_models/seedance.py,sha256=ni7LtXn4jTn5wX2NtcWDMj5Eea8LoP1QLYgwSx_GvBs,9014
@@ -14,18 +14,20 @@ media_agent_mcp/audio/speed_controller.py,sha256=IS9glznrTtbNR1ZHmaJe8Js545ttbWO
 media_agent_mcp/audio/tts.py,sha256=5lWZLXm56OUzUQYXAOu0sLUCp9FNDBrlrtvzQScXgHU,22524
 media_agent_mcp/be/README.md,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 media_agent_mcp/be/__init__.py,sha256=pyMpzXQGyOOjKlxKfYZur8EmoJcNIDQSM6Zgp63HPVE,97
-media_agent_mcp/be/app.py,sha256=_Tq7LJ9BOWcELvebYnHKCM42jY6e3fCXDtm-kDfTDvw,945
+media_agent_mcp/be/app.py,sha256=XdTI6fGELc083H8xEYGAcX1PJ6ISgWFu9R9C8i74_lo,1046
 media_agent_mcp/be/pyproject.toml,sha256=2c-vG61OpjH9Z2lz2aRoiDR047A-bOv3tFYKdxKXX68,199
-media_agent_mcp/be/routes_media.py,sha256=SsLNi0L3b8bbQpJV3sEiAwJw3S5PmjeCYnpQtVCjSGE,14454
+media_agent_mcp/be/routes_media.py,sha256=yBDQkOx5t7v9gkSoxGnaE4Okoia3SueX8m2N5ljKpeE,14623
+media_agent_mcp/be/routes_omni_human.py,sha256=SSmiGLvSJAQniwg-oODkoQnANL5iCuAKnwDTVQsPSlA,1097
 media_agent_mcp/be/routes_subtitles.py,sha256=jZYsPqyPqvIyuFyCnh7-EUpqjz-syqeqBUlhwwX0eLA,6833
 media_agent_mcp/be/utils.py,sha256=YQB_Fzwz0jhfJCOEHtahrv33VQuZZ3-_v5B8qGt3gkY,8027
 media_agent_mcp/be/uv.lock,sha256=U49x5h0EU0Sd2dn3EVvBbppM-XiujHk3Suw_z2U94SY,23502
-media_agent_mcp/be/__pycache__/__init__.cpython-312.pyc,sha256=ypxitYvyIAWOULz0FiRF3Wr_tlx4wKjUyTjnv-fKgsk,232
-media_agent_mcp/be/__pycache__/app.cpython-312.pyc,sha256=FCJLa895c4fIVJaQYDn4ipv3Spto5__lW-374cnwRNk,26673
-media_agent_mcp/be/__pycache__/routes_media.cpython-312.pyc,sha256=CQfwVSouBToVKnIoS1ZitthS_zjWhT97h0KOFIlO7gs,11094
+media_agent_mcp/be/__pycache__/__init__.cpython-312.pyc,sha256=5vVeA7KwnQ_acE3Mdjd36WbzeqRjhO6umus1RyAoxO4,225
+media_agent_mcp/be/__pycache__/app.cpython-312.pyc,sha256=QK7vq-fgrSYTb0laVfU3Yx_lug6CVVOR2KhgXtgpjFU,1401
+media_agent_mcp/be/__pycache__/routes_media.cpython-312.pyc,sha256=vVIANk-HcRxYGftPDh7XeUywHrk3cGXdGe0ckyOhytw,16374
 media_agent_mcp/be/__pycache__/routes_omni.cpython-312.pyc,sha256=ZwJctEAhdE74byCLlJ_F23j0pfVcIeLbvtKeygARdr0,12443
-media_agent_mcp/be/__pycache__/routes_subtitles.cpython-312.pyc,sha256=5Fy7RKhr535LpAj4UVT8bBAipSwgmTm1Q2b_1_evzPU,7916
-media_agent_mcp/be/__pycache__/utils.cpython-312.pyc,sha256=jsHMVf1DBvQa2gHPF7_OCIFnL2tAyUXBX5H_XTKXSyA,10161
+media_agent_mcp/be/__pycache__/routes_omni_human.cpython-312.pyc,sha256=VgJeY1aneYy4P0GEiyhpSUgTPLQf4664YeRXNpZ9dns,1485
+media_agent_mcp/be/__pycache__/routes_subtitles.cpython-312.pyc,sha256=W931dzRYi1Dk-YhWdLXfOUJE2Zjj741Fehxd_ZuNtfs,7883
+media_agent_mcp/be/__pycache__/utils.cpython-312.pyc,sha256=EKoj5x6Bt8Nv80tjL06ElO_9ZkvUBKvECzEEIKZ0oI0,11450
 media_agent_mcp/be/fonts/en/EduNSWACTCursive-VariableFont_wght.ttf,sha256=1SSZL9RXj58M7uJ_F68fRR1VQqxF6ZWK7mtoRCU-0hU,663016
 media_agent_mcp/be/fonts/en/MozillaText-VariableFont_wght.ttf,sha256=_hP_0MLtUZ6o1vG9yKaCNO0HpxXDET7DzzwicmUCSSo,84740
 media_agent_mcp/be/fonts/en/Roboto_Condensed-Regular.ttf,sha256=2Ob3xaE1bUHdgrDAAhn-vSHHkcxc0ez4oDI8vq_mSWw,145908
@@ -40,11 +42,12 @@ media_agent_mcp/media_selectors/video_selector.py,sha256=yuPbZMRm7fM0lyhoHaVN0-g
 media_agent_mcp/storage/__init__.py,sha256=eio7ZiSeLjCxICSZwZisiR7wKJfXlT2PV7aDEQKBUQQ,215
 media_agent_mcp/storage/tos_client.py,sha256=t2I-q63tMBahm4Ze1VRiN2Kn-VJGAjgqI0HAzJEJr7Y,3138
 media_agent_mcp/video/__init__.py,sha256=tfz22XEeFSeuKa3AggYCE0vCDt4IwXRCKW6avofyUsY,325
+media_agent_mcp/video/omni_human.py,sha256=vXGQM9pjbKdyhUkXVnsyLIfTobXokOa-PEZOz17yOpo,3908
 media_agent_mcp/video/processor.py,sha256=twfqmN5DbVryjDawZUcqTUcnglcBJYpUbAnApqHgD0c,12787
 media_agent_mcp/video/stack.py,sha256=pyoJiJ9NhU1tjy2l3kARI9sWFoC00Fj97psxYOBi2NU,1736
 media_agent_mcp/video/subtitle.py,sha256=TlrWVhWJqYTUJpnVz7eccwMAn8ixfrRzRxS6ETMY-DM,16323
-media_agent_mcp-2.7.0.dist-info/METADATA,sha256=LFjhceAxy7-5kSQoeuYu0MHb5aX_aA2yp1tJgegWjW4,11305
-media_agent_mcp-2.7.0.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
-media_agent_mcp-2.7.0.dist-info/entry_points.txt,sha256=qhOUwR-ORVf9GO7emhhl7Lgd6MISgqbZr8bEuSH_VdA,70
-media_agent_mcp-2.7.0.dist-info/top_level.txt,sha256=WEa0YfchpTxZgiKn8gdxYgs-dir5HepJaTOrxAGx9nY,16
-media_agent_mcp-2.7.0.dist-info/RECORD,,
+media_agent_mcp-2.7.2.dist-info/METADATA,sha256=ZaP4F-KrJm6eOAnNxnUcO2tzx3dnEibwavZNr0J-nqs,11337
+media_agent_mcp-2.7.2.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
+media_agent_mcp-2.7.2.dist-info/entry_points.txt,sha256=qhOUwR-ORVf9GO7emhhl7Lgd6MISgqbZr8bEuSH_VdA,70
+media_agent_mcp-2.7.2.dist-info/top_level.txt,sha256=WEa0YfchpTxZgiKn8gdxYgs-dir5HepJaTOrxAGx9nY,16
+media_agent_mcp-2.7.2.dist-info/RECORD,,