update

infer.py

@@ -1,36 +1,29 @@
-from fastapi import FastAPI, File, UploadFile, HTTPException
-from fastapi.responses import FileResponse
+import gradio as gr
 from gradio_client import Client
 import os
 import tempfile
 
-app = FastAPI()
-
-# Crea il client Gradio
-client = Client("http://127.0.0.1:6969/")
+# Crea il client Gradio per l'inferenza
+inference_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.")
+def process_audio(audio_file):
+    if audio_file is None:
+        return None, "Nessun file audio caricato."
+    
+    # Ottieni il percorso temporaneo del file caricato
+    input_file = audio_file.name
     
-    # 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_filename = f"generated_{os.path.basename(input_file)}"
         output_path = os.path.join(output_folder, output_filename)
 
         # Esegui la predizione
-        result = client.predict(
+        result = inference_client.predict(
             -24,  # Pece
             0,    # Raggio del filtro
             0,    # Rapporto feature di ricerca
@@ -50,14 +43,22 @@ async def process_audio(file: UploadFile = File(...)):
             api_name="/run_infer_script"
         )
 
-        # Restituisci il file elaborato
-        return FileResponse(output_path, media_type="audio/wav", filename=output_filename)
+        return output_path, f"Elaborazione completata. File salvato in {output_path}"
 
     except Exception as e:
-        raise HTTPException(status_code=500, detail=f"Errore durante l'elaborazione: {str(e)}")
+        return None, f"Errore durante l'elaborazione: {str(e)}"
 
-    finally:
-        # Pulisci il file temporaneo
-        os.unlink(input_file)
+# Creazione dell'interfaccia Gradio
+iface = gr.Interface(
+    fn=process_audio,
+    inputs=gr.Audio(type="filepath", label="Carica file audio (WAV o MP3)"),
+    outputs=[
+        gr.Audio(type="filepath", label="Audio elaborato"),
+        gr.Textbox(label="Messaggio")
+    ],
+    title="Elaborazione Audio con Applio",
+    description="Carica un file audio WAV o MP3 per elaborarlo con Applio."
+)
 
-# Non è necessario includere il blocco if __name__ == "__main__" in uno spazio Hugging Face
+# Avvio dell'interfaccia
+iface.launch()