Create infer.py

infer.py

@@ -0,0 +1,63 @@
+from fastapi import FastAPI, File, UploadFile, HTTPException
+from fastapi.responses import FileResponse
+from gradio_client import Client
+import os
+import tempfile
+
+app = FastAPI()
+
+# Crea il client Gradio
+client = Client("http://127.0.0.1:6969/")
+
+# Cartella di output in assets
+output_folder = "assets/generated"
+os.makedirs(output_folder, exist_ok=True)
+
+@app.post("/process_audio/")
+async def process_audio(file: UploadFile = File(...)):
+    # Verifica l'estensione del file
+    if not file.filename.lower().endswith(('.wav', '.mp3')):
+        raise HTTPException(status_code=400, detail="Solo file WAV e MP3 sono supportati.")
+    
+    # Salva temporaneamente il file caricato
+    with tempfile.NamedTemporaryFile(delete=False, suffix=os.path.splitext(file.filename)[1]) as temp_file:
+        temp_file.write(await file.read())
+        input_file = temp_file.name
+
+    try:
+        # Prepara il percorso di output
+        output_filename = f"generated_{os.path.basename(file.filename)}"
+        output_path = os.path.join(output_folder, output_filename)
+
+        # Esegui la predizione
+        result = client.predict(
+            -24,  # Pece
+            0,    # Raggio del filtro
+            0,    # Rapporto feature di ricerca
+            0,    # Inviluppo del volume
+            0,    # Proteggi le consonanti sorde
+            1,    # Lunghezza del luppolo
+            "pm", # Algoritmo di estrazione del passo
+            input_file,  # Percorso del file audio di input
+            output_path, # Percorso del file audio di output
+            "logs/master/master.pth", # Modello vocale
+            "logs/master/added_IVF124_Flat_nprobe_1_master-v2_v2.index", # File di indice
+            True, # Dividere l'audio
+            True, # Sintonizzazione automatica
+            True, # Audio pulito
+            0,    # Forza pulita
+            "WAV", # Export Format
+            api_name="/run_infer_script"
+        )
+
+        # Restituisci il file elaborato
+        return FileResponse(output_path, media_type="audio/wav", filename=output_filename)
+
+    except Exception as e:
+        raise HTTPException(status_code=500, detail=f"Errore durante l'elaborazione: {str(e)}")
+
+    finally:
+        # Pulisci il file temporaneo
+        os.unlink(input_file)
+
+# Non è necessario includere il blocco if __name__ == "__main__" in uno spazio Hugging Face