File size: 3,772 Bytes
c63c39e 2e6074b c63c39e 2e6074b c63c39e 2e6074b c63c39e 2e6074b c63c39e 2e6074b c63c39e 2e6074b c63c39e 2e6074b c63c39e | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 | import json
import os
import sys
# Setup path
current_dir = os.path.dirname(os.path.abspath(__file__))
clipping_dir = os.path.dirname(current_dir)
if clipping_dir not in sys.path:
sys.path.append(clipping_dir)
try:
from fastapi.testclient import TestClient
from main import app
from routers.video import ORIGINALS_DIR
except ImportError as e:
print(f"❌ Error: Missing dependencies. Please run 'pip install -r requirements.txt'")
print(f"Details: {e}")
sys.exit(1)
client = TestClient(app)
def create_10s_mock_video(filename="test_10s_video.mp4"):
path = os.path.join(ORIGINALS_DIR, filename)
if not os.path.exists(path):
import imageio_ffmpeg
import subprocess
exe = imageio_ffmpeg.get_ffmpeg_exe()
subprocess.run([
exe, '-f', 'lavfi', '-i', 'testsrc=duration=10:size=1280x720:rate=30',
'-f', 'lavfi', '-i', 'sine=frequency=1000:duration=10',
'-c:v', 'libx264', '-t', '10', path, '-y'
], capture_output=True)
return filename
def test_10s_workflow():
video_name = create_10s_mock_video()
# اختبار 1: استخدام وضع الكلمات التقليدي (Word Mode) مع بريسيت gaming
transcription_words = [
{"word": "Hello", "start": 0.0, "end": 1.0},
{"word": "this", "start": 1.0, "end": 2.0},
{"word": "is", "start": 2.0, "end": 3.0},
{"word": "a", "start": 3.0, "end": 4.0},
{"word": "test", "start": 4.0, "end": 5.0}
]
# اختبار 2: استخدام وضع الجمل والمزامنة اليدوية (Sentence Mode) مع بريسيت storyteller
transcription_sentences = [
{
"text": "This is a test of the storyteller mode",
"start": 5.0,
"end": 10.0,
"words": [
{"word": "This", "start": 5.0, "end": 6.0},
{"word": "is", "start": 6.0, "end": 7.0},
{"word": "a", "start": 7.0, "end": 8.0},
{"word": "test", "start": 8.0, "end": 9.0},
{"word": "of", "start": 9.0, "end": 9.5},
{"word": "the", "start": 9.5, "end": 10.0}
]
}
]
timestamps = [{"start_time": 0, "end_time": 10}]
# تنفيذ الاختبار الأول (Word Mode)
print(f"\n--- Running Test 1: Word Mode with 'gaming' preset ---")
process_data_1 = {
"original_filename": video_name,
"aspect_ratio": "9:16",
"style": "cinematic",
"subtitle_style": "gaming",
"transcription": json.dumps(transcription_words),
"timestamps": json.dumps(timestamps)
}
resp1 = client.post("/api/video/process-saved", data=process_data_1)
if resp1.status_code == 200:
print(f"✅ Word Mode Success! Task ID: {resp1.json()['task_id']}")
else:
print(f"❌ Word Mode Failed: {resp1.text}")
# تنفيذ الاختبار الثاني (Sentence Mode - المزامنة اليدوية)
print(f"\n--- Running Test 2: Sentence Mode with 'storyteller' preset ---")
process_data_2 = {
"original_filename": video_name,
"aspect_ratio": "9:16",
"style": "cinematic",
"subtitle_style": "storyteller",
"transcription": json.dumps(transcription_sentences),
"timestamps": json.dumps(timestamps)
}
resp2 = client.post("/api/video/process-saved", data=process_data_2)
if resp2.status_code == 200:
result = resp2.json()
print(f"✅ Sentence Mode Success! Task ID: {result['task_id']}")
print(f"✅ Clip URL: {result['clips_urls'][0]}")
else:
print(f"❌ Sentence Mode Failed: {resp2.text}")
if __name__ == "__main__":
test_10s_workflow()
|