jiadisu commited on
Commit
fc5f622
·
1 Parent(s): 7c8ec6f
Files changed (1) hide show
  1. app.py +11 -3
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] LLM rewrite failed: {e}, using original prompt")
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
- raise gr.Error(result["error"])
 
77
 
78
  video_path = result["video_path"]
79
  if not video_path or not os.path.isfile(video_path):
80
- raise gr.Error("Video file not found.")
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