Sada8888 commited on
Commit
fcb05eb
·
verified ·
1 Parent(s): 1a9e883

Update yml/generate-audio.py

Browse files
Files changed (1) hide show
  1. yml/generate-audio.py +14 -4
yml/generate-audio.py CHANGED
@@ -1,10 +1,9 @@
1
  import os, sys, requests, time
2
  from gradio_client import Client, handle_file
3
 
4
- # --- مقادیر مستقیما از سمت Flask تزریق می‌شوند ---
5
  AUDIO_TYPE = '___AUDIO_TYPE___'
6
  NEGATIVE_PROMPT = '''___NEGATIVE_PROMPT___'''
7
- # -----------------------------------------------
8
 
9
  prompt = os.environ.get('PROMPT', '')
10
  try: seed = int(os.environ.get('SEED', '-1'))
@@ -25,13 +24,24 @@ def report_failure(error_msg):
25
  "run_id": run_id,
26
  "error": error_msg,
27
  "event_type": "generate-audio",
28
- "client_payload": {"prompt": prompt, "duration": duration, "run_id": run_id, "space_url": space_url},
 
 
 
 
 
 
 
 
 
 
 
29
  "github_run_id": github_run_id
30
  },
31
  timeout=15
32
  )
33
  except Exception as e:
34
- pass
35
 
36
  try:
37
  print('1. Connecting to Audio Space...')
 
1
  import os, sys, requests, time
2
  from gradio_client import Client, handle_file
3
 
4
+ # --- Variables injected by Flask ---
5
  AUDIO_TYPE = '___AUDIO_TYPE___'
6
  NEGATIVE_PROMPT = '''___NEGATIVE_PROMPT___'''
 
7
 
8
  prompt = os.environ.get('PROMPT', '')
9
  try: seed = int(os.environ.get('SEED', '-1'))
 
24
  "run_id": run_id,
25
  "error": error_msg,
26
  "event_type": "generate-audio",
27
+ "client_payload": {
28
+ "data": {
29
+ "prompt": prompt,
30
+ "duration": duration,
31
+ "seed": seed,
32
+ "audio_type": AUDIO_TYPE,
33
+ "negative_prompt": NEGATIVE_PROMPT,
34
+ "file_url": file_url,
35
+ "run_id": run_id,
36
+ "space_url": space_url
37
+ }
38
+ },
39
  "github_run_id": github_run_id
40
  },
41
  timeout=15
42
  )
43
  except Exception as e:
44
+ print(f"Failed to report failure: {e}")
45
 
46
  try:
47
  print('1. Connecting to Audio Space...')