victor HF staff commited on
Commit
36bd0a9
1 Parent(s): e2dc219

refactor: Improve ffmpeg command extraction from LLM response

Browse files
Files changed (1) hide show
  1. app.py +12 -1
app.py CHANGED
@@ -138,7 +138,18 @@ YOUR FFMPEG COMMAND:
138
  top_p=top_p,
139
  max_tokens=2048
140
  )
141
- command = completion.choices[0].message.content.replace("\n", "")
 
 
 
 
 
 
 
 
 
 
 
142
 
143
  # remove output.mp4 with the actual output file path
144
  command = command.replace("output.mp4", "")
 
138
  top_p=top_p,
139
  max_tokens=2048
140
  )
141
+ content = completion.choices[0].message.content
142
+ # Extract command from code block if present
143
+ if "```" in content:
144
+ # Find content between ```sh or ```bash and the next ```
145
+ import re
146
+ command = re.search(r'```(?:sh|bash)?\n(.*?)\n```', content, re.DOTALL)
147
+ if command:
148
+ command = command.group(1).strip()
149
+ else:
150
+ command = content.replace("\n", "")
151
+ else:
152
+ command = content.replace("\n", "")
153
 
154
  # remove output.mp4 with the actual output file path
155
  command = command.replace("output.mp4", "")