Spaces:
Sleeping
Sleeping
Update Hugging Face Space to output JSON and enhance API documentation
Browse files
app.py
CHANGED
@@ -93,12 +93,18 @@ def download_video(url_string: str, temp_dir: str) -> str | None:
|
|
93 |
return None
|
94 |
|
95 |
|
96 |
-
def process_video_input(input_string: str) -> str:
|
97 |
"""
|
98 |
-
Processes the video (from URL or local file path) and returns its transcription status.
|
99 |
"""
|
100 |
if not input_string:
|
101 |
-
return
|
|
|
|
|
|
|
|
|
|
|
|
|
102 |
|
103 |
video_path_to_process = None
|
104 |
created_temp_dir = None # To store path of temp directory if created for download
|
@@ -116,13 +122,25 @@ def process_video_input(input_string: str) -> str:
|
|
116 |
# Error message is already printed by download_video or this block
|
117 |
print(f"Failed to download or locate video from URL: {input_string}")
|
118 |
# Cleanup is handled in finally, so just return error
|
119 |
-
return
|
|
|
|
|
|
|
|
|
|
|
|
|
120 |
|
121 |
elif os.path.exists(input_string):
|
122 |
print(f"Input is a local file path: {input_string}")
|
123 |
video_path_to_process = input_string
|
124 |
else:
|
125 |
-
return
|
|
|
|
|
|
|
|
|
|
|
|
|
126 |
|
127 |
if video_path_to_process:
|
128 |
print(f"Processing video: {video_path_to_process}")
|
@@ -155,28 +173,63 @@ def process_video_input(input_string: str) -> str:
|
|
155 |
# the lookup `modal.Function.lookup("whisper-transcriber", "transcribe_video_audio")` is standard.
|
156 |
# If it was deployed as part of the default app, then just "transcribe_video_audio" might work.
|
157 |
# Given the deployment log, the first lookup should be correct.
|
158 |
-
return
|
|
|
|
|
|
|
|
|
|
|
|
|
159 |
|
160 |
print("Calling Modal function for transcription...")
|
161 |
# Using .remote() for asynchronous execution, .call() for synchronous
|
162 |
# For Gradio, synchronous (.call()) might be simpler to handle the response directly.
|
163 |
transcription = f.remote(video_bytes_content) # Use .remote() for Modal function call
|
164 |
print(f"Received transcription from Modal: {transcription[:100]}...")
|
165 |
-
return
|
|
|
|
|
|
|
|
|
|
|
166 |
except FileNotFoundError:
|
167 |
print(f"Error: Video file not found at {video_path_to_process} before sending to Modal.")
|
168 |
-
return
|
|
|
|
|
|
|
|
|
|
|
|
|
169 |
except modal.Error as e: # Using modal.Error as the base Modal exception
|
170 |
print(f"Modal specific error: {e}")
|
171 |
-
return
|
|
|
|
|
|
|
|
|
|
|
|
|
172 |
except Exception as e:
|
173 |
print(f"An unexpected error occurred while calling Modal: {e}")
|
174 |
import traceback
|
175 |
traceback.print_exc()
|
176 |
-
return
|
|
|
|
|
|
|
|
|
|
|
|
|
177 |
else:
|
178 |
# This case should ideally be caught by earlier checks
|
179 |
-
return
|
|
|
|
|
|
|
|
|
|
|
|
|
180 |
|
181 |
finally:
|
182 |
if created_temp_dir and os.path.exists(created_temp_dir):
|
@@ -192,10 +245,14 @@ api_interface = gr.Interface(
|
|
192 |
fn=process_video_input,
|
193 |
inputs=gr.Textbox(label="Video URL or Local File Path for Transcription",
|
194 |
placeholder="Enter YouTube URL, direct video URL (.mp4, .mov, etc.), or local file path..."),
|
195 |
-
outputs="
|
196 |
title="Video Transcription API",
|
197 |
-
description="Provide a video URL or local file path to get its audio transcription
|
198 |
-
allow_flagging="never"
|
|
|
|
|
|
|
|
|
199 |
)
|
200 |
|
201 |
# Gradio Interface for a simple user-facing demo
|
|
|
93 |
return None
|
94 |
|
95 |
|
96 |
+
def process_video_input(input_string: str) -> Dict[str, Any]:
|
97 |
"""
|
98 |
+
Processes the video (from URL or local file path) and returns its transcription status as a JSON object.
|
99 |
"""
|
100 |
if not input_string:
|
101 |
+
return {
|
102 |
+
"status": "error",
|
103 |
+
"error_details": {
|
104 |
+
"message": "No video URL or file path provided.",
|
105 |
+
"input_received": input_string
|
106 |
+
}
|
107 |
+
}
|
108 |
|
109 |
video_path_to_process = None
|
110 |
created_temp_dir = None # To store path of temp directory if created for download
|
|
|
122 |
# Error message is already printed by download_video or this block
|
123 |
print(f"Failed to download or locate video from URL: {input_string}")
|
124 |
# Cleanup is handled in finally, so just return error
|
125 |
+
return {
|
126 |
+
"status": "error",
|
127 |
+
"error_details": {
|
128 |
+
"message": "Failed to download video from URL.",
|
129 |
+
"input_received": input_string
|
130 |
+
}
|
131 |
+
}
|
132 |
|
133 |
elif os.path.exists(input_string):
|
134 |
print(f"Input is a local file path: {input_string}")
|
135 |
video_path_to_process = input_string
|
136 |
else:
|
137 |
+
return {
|
138 |
+
"status": "error",
|
139 |
+
"error_details": {
|
140 |
+
"message": f"Input '{input_string}' is not a valid URL or an existing file path.",
|
141 |
+
"input_received": input_string
|
142 |
+
}
|
143 |
+
}
|
144 |
|
145 |
if video_path_to_process:
|
146 |
print(f"Processing video: {video_path_to_process}")
|
|
|
173 |
# the lookup `modal.Function.lookup("whisper-transcriber", "transcribe_video_audio")` is standard.
|
174 |
# If it was deployed as part of the default app, then just "transcribe_video_audio" might work.
|
175 |
# Given the deployment log, the first lookup should be correct.
|
176 |
+
return {
|
177 |
+
"status": "error",
|
178 |
+
"error_details": {
|
179 |
+
"message": "Could not find the deployed Modal function. Please check deployment status and name.",
|
180 |
+
"modal_function_name": "whisper-transcriber/transcribe_video_audio"
|
181 |
+
}
|
182 |
+
}
|
183 |
|
184 |
print("Calling Modal function for transcription...")
|
185 |
# Using .remote() for asynchronous execution, .call() for synchronous
|
186 |
# For Gradio, synchronous (.call()) might be simpler to handle the response directly.
|
187 |
transcription = f.remote(video_bytes_content) # Use .remote() for Modal function call
|
188 |
print(f"Received transcription from Modal: {transcription[:100]}...")
|
189 |
+
return {
|
190 |
+
"status": "success",
|
191 |
+
"data": {
|
192 |
+
"transcription": transcription
|
193 |
+
}
|
194 |
+
}
|
195 |
except FileNotFoundError:
|
196 |
print(f"Error: Video file not found at {video_path_to_process} before sending to Modal.")
|
197 |
+
return {
|
198 |
+
"status": "error",
|
199 |
+
"error_details": {
|
200 |
+
"message": "Video file disappeared before processing.",
|
201 |
+
"path_attempted": video_path_to_process
|
202 |
+
}
|
203 |
+
}
|
204 |
except modal.Error as e: # Using modal.Error as the base Modal exception
|
205 |
print(f"Modal specific error: {e}")
|
206 |
+
return {
|
207 |
+
"status": "error",
|
208 |
+
"error_details": {
|
209 |
+
"message": f"Error during Modal operation: {str(e)}",
|
210 |
+
"exception_type": type(e).__name__
|
211 |
+
}
|
212 |
+
}
|
213 |
except Exception as e:
|
214 |
print(f"An unexpected error occurred while calling Modal: {e}")
|
215 |
import traceback
|
216 |
traceback.print_exc()
|
217 |
+
return {
|
218 |
+
"status": "error",
|
219 |
+
"error_details": {
|
220 |
+
"message": f"Failed to get transcription: {str(e)}",
|
221 |
+
"exception_type": type(e).__name__
|
222 |
+
}
|
223 |
+
}
|
224 |
else:
|
225 |
# This case should ideally be caught by earlier checks
|
226 |
+
return {
|
227 |
+
"status": "error",
|
228 |
+
"error_details": {
|
229 |
+
"message": "No video available to process after input handling.",
|
230 |
+
"input_received": input_string
|
231 |
+
}
|
232 |
+
}
|
233 |
|
234 |
finally:
|
235 |
if created_temp_dir and os.path.exists(created_temp_dir):
|
|
|
245 |
fn=process_video_input,
|
246 |
inputs=gr.Textbox(label="Video URL or Local File Path for Transcription",
|
247 |
placeholder="Enter YouTube URL, direct video URL (.mp4, .mov, etc.), or local file path..."),
|
248 |
+
outputs=gr.JSON(label="API Response"),
|
249 |
title="Video Transcription API",
|
250 |
+
description="Provide a video URL or local file path to get its audio transcription. Output is JSON.",
|
251 |
+
allow_flagging="never",
|
252 |
+
examples=[
|
253 |
+
["https://www.youtube.com/watch?v=dQw4w9WgXcQ"],
|
254 |
+
["https://commondatastorage.googleapis.com/gtv-videos-bucket/sample/BigBuckBunny.mp4"]
|
255 |
+
]
|
256 |
)
|
257 |
|
258 |
# Gradio Interface for a simple user-facing demo
|