Update yml/generate-audio.py
Browse files- 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 |
-
# ---
|
| 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": {
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 29 |
"github_run_id": github_run_id
|
| 30 |
},
|
| 31 |
timeout=15
|
| 32 |
)
|
| 33 |
except Exception as e:
|
| 34 |
-
|
| 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...')
|