Update app.py

app.py

@@ -3,11 +3,21 @@ import uuid
 import shutil
 import secrets
 import traceback
+import torch  # Ditambahkan untuk melakukan patch sistem
 from fastapi import FastAPI, Depends, HTTPException, status, File, UploadFile, Form
 from fastapi.security import HTTPBasic, HTTPBasicCredentials
 from fastapi.responses import FileResponse
 from TTS.api import TTS
 
+# --- PATCH PYTORCH 2.6+ ---
+# Memaksa PyTorch untuk mengizinkan pemuatan objek kustom Coqui TTS
+original_torch_load = torch.load
+def patched_torch_load(*args, **kwargs):
+    kwargs['weights_only'] = False
+    return original_torch_load(*args, **kwargs)
+torch.load = patched_torch_load
+# --------------------------
+
 # --- KONFIGURASI FOLDER TEMP ---
 TEMP_DIR = "/tmp/coqui_data"
 AUDIO_DIR = "/tmp/audio_output"
@@ -20,8 +30,8 @@ os.environ["COQUI_TOS_AGREED"] = "1"
 
 app = FastAPI(
     title="PasBlast Dynamic Coqui API",
-    description="API Komplet dengan Dynamic Model Loading. Memanfaatkan 16GB RAM untuk menjalankan berbagai model secara leluasa.",
-    version="2.0.0"
+    description="API Komplet dengan Dynamic Model Loading dan PyTorch Patch. Memanfaatkan 16GB RAM untuk menjalankan berbagai model secara leluasa.",
+    version="2.0.1"
 )
 security = HTTPBasic()
 
@@ -119,26 +129,21 @@ def generate_tts(
     tts = get_tts_instance(model_name)
     output_path = os.path.join(AUDIO_DIR, f"tts_{uuid.uuid4().hex}.wav")
     
-    # Validasi dan auto-fallback speaker jika model membutuhkan speaker
     if hasattr(tts, "speakers") and tts.speakers:
         if speaker not in tts.speakers:
             print(f"Speaker '{speaker}' tidak valid untuk model {model_name}. Menggunakan default.")
             speaker = tts.speakers[0]
             
-    # Validasi language
     if hasattr(tts, "languages") and tts.languages:
         if language not in tts.languages:
             language = tts.languages[0]
         
     try:
         print(f"Memproses TTS menggunakan {model_name}...")
-        
-        # Eksekusi berdasarkan apakah model XTTS atau VITS/FastPitch biasa
         kwargs = {"text": text, "file_path": output_path}
         if speaker: kwargs["speaker"] = speaker
         if language: kwargs["language"] = language
         
-        # XTTS mendukung parameter tuning tambahan, model lama mungkin tidak
         if "xtts" in model_name.lower():
             kwargs.update({"split_sentences": True, "temperature": temperature, "length_penalty": length_penalty, "repetition_penalty": repetition_penalty})
             
@@ -152,7 +157,7 @@ def generate_tts(
 @app.post("/tts_voice_clone", tags=["Core Generation"])
 def generate_tts_voice_clone(
     text: str = Form(...),
-    model_name: str = Form("tts_models/multilingual/multi-dataset/xtts_v2", description="Harus model yang mendukung kloning (seperti xtts_v2 atau your_tts)"),
+    model_name: str = Form("tts_models/multilingual/multi-dataset/xtts_v2", description="Harus model yang mendukung kloning"),
     language: str = Form("id"),
     reference_audio: UploadFile = File(...),
     temperature: float = Form(0.75),
@@ -175,7 +180,6 @@ def generate_tts_voice_clone(
         tts.tts_to_file(**kwargs)
         return FileResponse(output_path, media_type="audio/wav", filename="pasblast_cloned.wav")
     except Exception as e:
-        error_trace = traceback.format_exc()
         raise HTTPException(status_code=500, detail=f"Gagal memproses Voice Cloning: {str(e)}")
 
 @app.post("/voice_conversion", tags=["Core Generation"])