fix: gradio cache error and moshi

gradio_app.py

@@ -9,6 +9,7 @@ import warnings
 import multiprocessing
 import concurrent.futures
 import urllib.request
+import pathlib
 
 try:
     from moshi.models.tts import TTSModel
@@ -26,11 +27,14 @@ from notebook_lm_kokoro import (
 os.environ["HOME"] = "/tmp/home"  # Prevent fallback to /root
 os.makedirs("/tmp/home", exist_ok=True)
 os.environ["GRADIO_TEMP_DIR"] = "/tmp/gradio"
-os.environ["HF_HOME"] = "/tmp/huggingface"
 os.environ["TRANSFORMERS_CACHE"] = "/tmp/huggingface/transformers"
-os.environ["XDG_CACHE_HOME"] = "/tmp/huggingface"
 os.environ["TORCH_HOME"] = "/tmp/torch"
 os.environ["MOSHI_CACHE_DIR"] = "/tmp/moshi"
+os.environ["XDG_CACHE_HOME"] = "/tmp/huggingface"  # already good
+os.environ["HF_HOME"] = "/tmp/huggingface"         # already good
+
+# Patch GRADIO’s cache path resolution (so it doesn't default to '/.cache')
+gr.paths.default_cache_dir = lambda: pathlib.Path("/tmp/huggingface/gradio")
 
 
 warnings.filterwarnings("ignore")
@@ -126,11 +130,7 @@ def process_pdf(pdf_file, speaker1_voice, speaker2_voice, kyutai_voice1, kyutai_
         if tts_engine == "kyutai":
             result = generate_audio_kyutai(transcript, kyutai_voice1, kyutai_voice2, audio_path)
         else:
-            with concurrent.futures.ProcessPoolExecutor(max_workers=NUM_WORKERS) as executor:
-                result = executor.submit(
-                    generate_audio_from_script_with_voices,
-                    transcript, speaker1_voice, speaker2_voice, audio_path
-                ).result()
+            result = generate_audio_from_script_with_voices(transcript, speaker1_voice, speaker2_voice, audio_path)
 
         return ("Process complete!", result) if result else ("Error generating audio", None)
     except Exception as e:

requirements.txt

@@ -6,5 +6,5 @@ numpy
 openai
 ipython
 gradio>=4.0.0
-moshi>=0.2.4
+git+https://[email protected]/kyutai-labs/moshi.git#egg=moshi&subdirectory=moshi
 sphn>=0.2.0