muhtasham commited on
Commit
78d9435
·
1 Parent(s): 67b74cb
Files changed (1) hide show
  1. app.py +24 -5
app.py CHANGED
@@ -214,7 +214,9 @@ def check_api_health():
214
 
215
  if cpu_percent > 90 or memory_percent > 90:
216
  logger.warning(f"High resource usage detected - CPU: {cpu_percent}%, Memory: {memory_percent}%")
217
-
 
 
218
  return True
219
 
220
  except requests.exceptions.RequestException as e:
@@ -262,10 +264,17 @@ def transcribe_youtube(url, return_timestamps, generate_subs, chunk_length_s=30,
262
  metadata = result.get("metadata", {})
263
  logger.info(f"Transcription metadata: {metadata}")
264
 
265
- # Format response with segments
266
  formatted_result = {
267
  "text": result["transcription"]["text"],
268
- "segments": result["transcription"]["segments"] if return_timestamps else None
 
 
 
 
 
 
 
269
  }
270
 
271
  # Generate subtitles if requested
@@ -303,6 +312,8 @@ def transcribe(inputs, return_timestamps, generate_subs, chunk_length_s=30, batc
303
  "chunk_length_s": chunk_length_s,
304
  "batch_size": batch_size
305
  }
 
 
306
 
307
  # Send request to API
308
  response = requests.post(
@@ -317,10 +328,17 @@ def transcribe(inputs, return_timestamps, generate_subs, chunk_length_s=30, batc
317
  metadata = result.get("metadata", {})
318
  logger.info(f"Transcription metadata: {metadata}")
319
 
320
- # Format response with segments
321
  formatted_result = {
322
  "text": result["transcription"]["text"],
323
- "segments": result["transcription"]["segments"] if return_timestamps else None
 
 
 
 
 
 
 
324
  }
325
 
326
  # Generate subtitles if requested
@@ -417,3 +435,4 @@ with demo:
417
  logger.info("Starting Gradio interface")
418
  demo.queue().launch(ssr_mode=False)
419
 
 
 
214
 
215
  if cpu_percent > 90 or memory_percent > 90:
216
  logger.warning(f"High resource usage detected - CPU: {cpu_percent}%, Memory: {memory_percent}%")
217
+
218
+ logger.info("API health check passed successfully")
219
+
220
  return True
221
 
222
  except requests.exceptions.RequestException as e:
 
264
  metadata = result.get("metadata", {})
265
  logger.info(f"Transcription metadata: {metadata}")
266
 
267
+ # Format response with segments (without id)
268
  formatted_result = {
269
  "text": result["transcription"]["text"],
270
+ "segments": [
271
+ {
272
+ "start": segment["start"],
273
+ "end": segment["end"],
274
+ "text": segment["text"]
275
+ }
276
+ for segment in result["transcription"]["segments"]
277
+ ] if return_timestamps else None
278
  }
279
 
280
  # Generate subtitles if requested
 
312
  "chunk_length_s": chunk_length_s,
313
  "batch_size": batch_size
314
  }
315
+
316
+ logger.info(f"Sending request to API with parameters: {params}")
317
 
318
  # Send request to API
319
  response = requests.post(
 
328
  metadata = result.get("metadata", {})
329
  logger.info(f"Transcription metadata: {metadata}")
330
 
331
+ # Format response with segments (without id)
332
  formatted_result = {
333
  "text": result["transcription"]["text"],
334
+ "segments": [
335
+ {
336
+ "start": segment["start"],
337
+ "end": segment["end"],
338
+ "text": segment["text"]
339
+ }
340
+ for segment in result["transcription"]["segments"]
341
+ ] if return_timestamps else None
342
  }
343
 
344
  # Generate subtitles if requested
 
435
  logger.info("Starting Gradio interface")
436
  demo.queue().launch(ssr_mode=False)
437
 
438
+