Initial commit with FastAPI + Gradio app

app.py

@@ -2,7 +2,7 @@ import io
 import tempfile
 from fastapi import FastAPI, HTTPException, File, UploadFile
 from speechbrain.inference import SpeakerRecognition
-from fastapi.openapi.docs import get_swagger_ui_html
+import gradio as gr
 
 # Initialize the speaker verification model
 verification = SpeakerRecognition.from_hparams(
@@ -55,12 +55,27 @@ async def compare_voices_api(file1: UploadFile = File(...), file2: UploadFile =
     except Exception as e:
         raise HTTPException(status_code=400, detail=str(e))
 
-# Custom route to expose Swagger UI
-@app.get("/docs")
-def custom_swagger_ui_html():
-    return get_swagger_ui_html(openapi_url=app.openapi_url, title="API Docs")
+# Gradio interface function
+def gradio_interface():
+    return gr.Interface(
+        fn=compare_voices_api,  # FastAPI function is wrapped here
+        inputs=[
+            gr.Audio(type="filepath", label="First Audio File"),  # Audio file input
+            gr.Audio(type="filepath", label="Second Audio File")  # Audio file input
+        ],
+        outputs="json",  # Output as JSON
+        live=False  # No live interface, just the API
+    )
 
-# Running the FastAPI app
+# Launch Gradio interface
+@app.on_event("startup")
+async def startup():
+    gr.Interface(fn=compare_voices_api, inputs=[
+        gr.Audio(type="filepath", label="First Audio File"),  # Audio file input
+        gr.Audio(type="filepath", label="Second Audio File")  # Audio file input
+    ], outputs="json", live=False).launch(share=True, inline=True)
+
+# Running the FastAPI app with Gradio
 if __name__ == "__main__":
     import uvicorn
-    uvicorn.run(app, host="127.0.0.0", port=5000)
+    uvicorn.run(app, host="0.0.0.0", port=5000)