updated app
Browse files- gradio_app.py +19 -14
gradio_app.py
CHANGED
@@ -39,7 +39,8 @@ def generate_audio_from_script_with_voices(script, speaker1_voice, speaker2_voic
|
|
39 |
chosen_voice = voice_map.get(speaker, "af_heart")
|
40 |
print(f"Generating audio for {speaker} with voice '{chosen_voice}'...")
|
41 |
|
42 |
-
|
|
|
43 |
generator = pipeline(dialogue, voice=chosen_voice)
|
44 |
|
45 |
segment_audio = []
|
@@ -152,15 +153,10 @@ def create_gradio_app():
|
|
152 |
)
|
153 |
speaker2_voice = gr.Dropdown(
|
154 |
choices=["af_nicole", "af_heart", "bf_emma"],
|
155 |
-
value="
|
156 |
label="Speaker 2 Voice"
|
157 |
)
|
158 |
|
159 |
-
provider = gr.Radio(
|
160 |
-
choices=["openai", "openrouter"],
|
161 |
-
value="openrouter",
|
162 |
-
label="API Provider"
|
163 |
-
)
|
164 |
|
165 |
with gr.Group():
|
166 |
provider = gr.Radio(
|
@@ -173,13 +169,14 @@ def create_gradio_app():
|
|
173 |
label="API Key",
|
174 |
placeholder="Enter your API key here...",
|
175 |
type="password",
|
176 |
-
|
177 |
)
|
178 |
|
179 |
openrouter_base = gr.Textbox(
|
180 |
label="OpenRouter Base URL (optional)",
|
181 |
placeholder="https://openrouter.ai/api/v1",
|
182 |
-
visible=False
|
|
|
183 |
)
|
184 |
|
185 |
# Show/hide OpenRouter base URL based on provider selection
|
@@ -207,9 +204,9 @@ def create_gradio_app():
|
|
207 |
# Examples section
|
208 |
gr.Examples(
|
209 |
examples=[
|
210 |
-
["sample.pdf", "af_heart", "af_nicole", "openrouter"],
|
211 |
],
|
212 |
-
inputs=[pdf_input, speaker1_voice, speaker2_voice, provider],
|
213 |
outputs=[status_output, audio_output],
|
214 |
fn=process_pdf,
|
215 |
cache_examples=True,
|
@@ -217,7 +214,14 @@ def create_gradio_app():
|
|
217 |
|
218 |
submit_btn.click(
|
219 |
fn=process_pdf,
|
220 |
-
inputs=[
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
221 |
outputs=[status_output, audio_output],
|
222 |
api_name="generate"
|
223 |
)
|
@@ -235,9 +239,10 @@ def create_gradio_app():
|
|
235 |
|
236 |
if __name__ == "__main__":
|
237 |
demo = create_gradio_app()
|
238 |
-
demo.queue(
|
239 |
server_name="0.0.0.0",
|
240 |
server_port=7860,
|
241 |
share=True,
|
242 |
-
debug=True
|
|
|
243 |
)
|
|
|
39 |
chosen_voice = voice_map.get(speaker, "af_heart")
|
40 |
print(f"Generating audio for {speaker} with voice '{chosen_voice}'...")
|
41 |
|
42 |
+
# Updated KPipeline initialization with explicit repo_id
|
43 |
+
pipeline = KPipeline(lang_code="a", repo_id="hexgrad/Kokoro-82M")
|
44 |
generator = pipeline(dialogue, voice=chosen_voice)
|
45 |
|
46 |
segment_audio = []
|
|
|
153 |
)
|
154 |
speaker2_voice = gr.Dropdown(
|
155 |
choices=["af_nicole", "af_heart", "bf_emma"],
|
156 |
+
value="bf_emma",
|
157 |
label="Speaker 2 Voice"
|
158 |
)
|
159 |
|
|
|
|
|
|
|
|
|
|
|
160 |
|
161 |
with gr.Group():
|
162 |
provider = gr.Radio(
|
|
|
169 |
label="API Key",
|
170 |
placeholder="Enter your API key here...",
|
171 |
type="password",
|
172 |
+
elem_classes="api-input"
|
173 |
)
|
174 |
|
175 |
openrouter_base = gr.Textbox(
|
176 |
label="OpenRouter Base URL (optional)",
|
177 |
placeholder="https://openrouter.ai/api/v1",
|
178 |
+
visible=False,
|
179 |
+
elem_classes="api-input"
|
180 |
)
|
181 |
|
182 |
# Show/hide OpenRouter base URL based on provider selection
|
|
|
204 |
# Examples section
|
205 |
gr.Examples(
|
206 |
examples=[
|
207 |
+
["sample.pdf", "af_heart", "af_nicole", "openrouter", "your-api-key-here", "https://openrouter.ai/api/v1"],
|
208 |
],
|
209 |
+
inputs=[pdf_input, speaker1_voice, speaker2_voice, provider, api_key, openrouter_base],
|
210 |
outputs=[status_output, audio_output],
|
211 |
fn=process_pdf,
|
212 |
cache_examples=True,
|
|
|
214 |
|
215 |
submit_btn.click(
|
216 |
fn=process_pdf,
|
217 |
+
inputs=[
|
218 |
+
pdf_input,
|
219 |
+
speaker1_voice,
|
220 |
+
speaker2_voice,
|
221 |
+
provider,
|
222 |
+
api_key,
|
223 |
+
openrouter_base
|
224 |
+
],
|
225 |
outputs=[status_output, audio_output],
|
226 |
api_name="generate"
|
227 |
)
|
|
|
239 |
|
240 |
if __name__ == "__main__":
|
241 |
demo = create_gradio_app()
|
242 |
+
demo.queue().launch(
|
243 |
server_name="0.0.0.0",
|
244 |
server_port=7860,
|
245 |
share=True,
|
246 |
+
debug=True,
|
247 |
+
pwa=True
|
248 |
)
|