Spaces:
Runtime error
Runtime error
jiadisu commited on
Commit ·
fc5f622
1
Parent(s): 7c8ec6f
update
Browse files
app.py
CHANGED
|
@@ -33,6 +33,8 @@ with open(_PROMPT_FILE, "r") as f:
|
|
| 33 |
|
| 34 |
def rewrite_prompt(user_prompt: str) -> str:
|
| 35 |
"""Rewrite user prompt into the model's required format via LLM API."""
|
|
|
|
|
|
|
| 36 |
try:
|
| 37 |
resp = _rewrite_client.chat.completions.create(
|
| 38 |
model=_REWRITE_MODEL,
|
|
@@ -44,9 +46,11 @@ def rewrite_prompt(user_prompt: str) -> str:
|
|
| 44 |
max_tokens=1024,
|
| 45 |
)
|
| 46 |
rewritten = resp.choices[0].message.content.strip()
|
|
|
|
|
|
|
| 47 |
return rewritten
|
| 48 |
except Exception as e:
|
| 49 |
-
print(f"[Rewrite]
|
| 50 |
return user_prompt
|
| 51 |
|
| 52 |
|
|
@@ -61,9 +65,11 @@ def run_generation(image, prompt, seed, seconds):
|
|
| 61 |
raise gr.Error("Please enter a text prompt.")
|
| 62 |
|
| 63 |
# Step 1: Rewrite prompt via LLM
|
|
|
|
| 64 |
rewritten = rewrite_prompt(prompt.strip())
|
| 65 |
|
| 66 |
# Step 2: Send to router
|
|
|
|
| 67 |
result = generate(
|
| 68 |
image=image,
|
| 69 |
video_prompt=rewritten,
|
|
@@ -73,13 +79,15 @@ def run_generation(image, prompt, seed, seconds):
|
|
| 73 |
)
|
| 74 |
|
| 75 |
if result["error"]:
|
| 76 |
-
|
|
|
|
| 77 |
|
| 78 |
video_path = result["video_path"]
|
| 79 |
if not video_path or not os.path.isfile(video_path):
|
| 80 |
-
|
| 81 |
|
| 82 |
status = f"Done. seed={result['seed']}"
|
|
|
|
| 83 |
return video_path, rewritten, status
|
| 84 |
|
| 85 |
|
|
|
|
| 33 |
|
| 34 |
def rewrite_prompt(user_prompt: str) -> str:
|
| 35 |
"""Rewrite user prompt into the model's required format via LLM API."""
|
| 36 |
+
print(f"[Rewrite] Starting rewrite, input length={len(user_prompt)} chars")
|
| 37 |
+
print(f"[Rewrite] User prompt: {user_prompt[:100]}...")
|
| 38 |
try:
|
| 39 |
resp = _rewrite_client.chat.completions.create(
|
| 40 |
model=_REWRITE_MODEL,
|
|
|
|
| 46 |
max_tokens=1024,
|
| 47 |
)
|
| 48 |
rewritten = resp.choices[0].message.content.strip()
|
| 49 |
+
print(f"[Rewrite] Done, output length={len(rewritten)} chars")
|
| 50 |
+
print(f"[Rewrite] Rewritten: {rewritten[:150]}...")
|
| 51 |
return rewritten
|
| 52 |
except Exception as e:
|
| 53 |
+
print(f"[Rewrite] FAILED: {e}, using original prompt")
|
| 54 |
return user_prompt
|
| 55 |
|
| 56 |
|
|
|
|
| 65 |
raise gr.Error("Please enter a text prompt.")
|
| 66 |
|
| 67 |
# Step 1: Rewrite prompt via LLM
|
| 68 |
+
print(f"[Generate] Request received: seed={seed} seconds={seconds} prompt={prompt[:50]!r}")
|
| 69 |
rewritten = rewrite_prompt(prompt.strip())
|
| 70 |
|
| 71 |
# Step 2: Send to router
|
| 72 |
+
print(f"[Generate] Sending to router ...")
|
| 73 |
result = generate(
|
| 74 |
image=image,
|
| 75 |
video_prompt=rewritten,
|
|
|
|
| 79 |
)
|
| 80 |
|
| 81 |
if result["error"]:
|
| 82 |
+
print(f"[Generate] Error from router: {result['error']}")
|
| 83 |
+
return None, rewritten, f"Error: {result['error']}"
|
| 84 |
|
| 85 |
video_path = result["video_path"]
|
| 86 |
if not video_path or not os.path.isfile(video_path):
|
| 87 |
+
return None, rewritten, "Error: Video file not found."
|
| 88 |
|
| 89 |
status = f"Done. seed={result['seed']}"
|
| 90 |
+
print(f"[Generate] Success: {video_path}")
|
| 91 |
return video_path, rewritten, status
|
| 92 |
|
| 93 |
|