Update app.py

app.py

@@ -7,7 +7,6 @@ from fastapi import FastAPI, File, UploadFile, Form, HTTPException
 
 app = FastAPI()
 
-# Voice choices from PersonaPlex repo
 VOICE_CHOICES = [
     "NATF0.pt", "NATF1.pt", "NATF2.pt", "NATF3.pt",
     "NATM0.pt", "NATM1.pt", "NATM2.pt", "NATM3.pt",
@@ -33,9 +32,9 @@ def run_offline_inference(input_path, text_prompt, voice_prompt, seed, output_wa
             prompt_file.write(text_prompt)
             cmd += ["--text-prompt", prompt_file.name]
         try:
-            subprocess.check_call(cmd, timeout=600)  # 10 min max per call (CPU safety)
+            subprocess.check_call(cmd, timeout=900)  # 15 min max (CPU can be slow)
         except subprocess.TimeoutExpired:
-            raise RuntimeError("Inference timed out — too slow on CPU?")
+            raise RuntimeError("Inference timed out — CPU is slow, try shorter input audio.")
         finally:
             if os.path.exists(prompt_file.name):
                 os.unlink(prompt_file.name)
@@ -46,7 +45,6 @@ def gradio_generate(input_audio, text_prompt, voice_prompt, seed):
     if input_audio is None:
         raise gr.Error("Please record or upload audio for Sofia to hear you!")
     
-    # Use default persona if user leaves it blank
     full_prompt = text_prompt.strip() or DEFAULT_PERSONA
     
     try:
@@ -54,8 +52,7 @@ def gradio_generate(input_audio, text_prompt, voice_prompt, seed):
             output_wav = os.path.join(tmpdir, "sofia_response.wav")
             output_json = os.path.join(tmpdir, "sofia_response.json")
             
-            # Show progress (Gradio will display this)
-            yield None, "Processing... Sofia is thinking (may take 1–3 minutes on CPU)..."
+            yield None, "Processing... Sofia is thinking (expect 1–5+ minutes on free CPU)..."
             
             run_offline_inference(input_audio, full_prompt, voice_prompt, seed, output_wav, output_json)
             
@@ -64,29 +61,32 @@ def gradio_generate(input_audio, text_prompt, voice_prompt, seed):
             
             yield output_wav, text
     except Exception as e:
-        raise gr.Error(f"Something went wrong: {str(e)}\n(Try shorter audio or check logs)")
+        raise gr.Error(f"Error: {str(e)}\n(Try shorter audio clips or check Space logs)")
 
-# Create Gradio interface
 with gr.Blocks(theme=gr.themes.Soft(primary_hue="pink", secondary_hue="purple")) as demo:
     gr.Markdown("# Sofia — Your PersonaPlex AI Companion")
     gr.Markdown(
-        "Record or upload audio → Sofia responds naturally! "
-        "**Note on CPU:** First response may take 2–5+ minutes (model loading). Subsequent ones faster. Be patient!"
+        "Record or upload short audio → Sofia responds!  \n"
+        "**CPU note:** First load takes time (model download + init). Responses: 1–5+ min. Use short inputs (5–15 sec)."
     )
     
     with gr.Row():
         with gr.Column():
             input_audio = gr.Audio(
-                sources=["microphone", "upload"], 
-                type="filepath", 
+                sources=["microphone", "upload"],
+                type="filepath",
                 label="Speak to Sofia (mic or upload WAV/MP3)",
-                waveform_options={"show_recording": True}
+                # Fixed: Use proper WaveformOptions class
+                waveform_options=gr.WaveformOptions(
+                    show_recording_waveform=True,   # shows waveform while recording
+                    show_controls=False             # optional: hides extra player buttons if wanted
+                )
             )
             text_prompt = gr.Textbox(
                 label="Custom Persona / Role for Sofia (optional)",
                 placeholder=DEFAULT_PERSONA,
                 lines=4,
-                value=""  # empty → uses default
+                value=""
             )
             voice_prompt = gr.Dropdown(
                 choices=VOICE_CHOICES,
@@ -99,23 +99,19 @@ with gr.Blocks(theme=gr.themes.Soft(primary_hue="pink", secondary_hue="purple"))
         with gr.Column():
             output_audio = gr.Audio(label="Sofia's Response (Audio)", autoplay=True)
             output_text = gr.Textbox(label="Sofia's Response (Text)", lines=6)
-            status = gr.Textbox(label="Status", interactive=False)
+            status = gr.Textbox(label="Status", interactive=False, value="Ready...")
 
-    # Use .submit() for better feedback on slow runs
     submit_btn.click(
         fn=gradio_generate,
         inputs=[input_audio, text_prompt, voice_prompt, seed],
         outputs=[output_audio, output_text],
-        api_name="generate_gradio"  # optional
     ).then(
-        lambda: "Done! Play above ↑",
+        lambda: "Done! Play Sofia's response above ↑",
         outputs=status
     )
 
-# Mount Gradio at root (cleaner URL: yourspace.hf.space directly shows UI)
 gr.mount_gradio_app(app, demo, path="/")
 
-# Keep the FastAPI /generate endpoint for external frontends
 @app.post("/generate")
 async def api_generate(
     input_audio: UploadFile = File(...),