Update app.py

app.py

@@ -1,31 +1,27 @@
-import gradio as gr
+import uvicorn
+from fastapi import FastAPI, File, UploadFile
 from func import detect_language_local, transcribe_and_diarize_audio
 
-def transcribe(file):
-    # # Сохраняем загруженный файл временно
-    # file_location = f"temp_{file.name}"
+app = FastAPI(docs_url="/")
 
-    # # Читаем содержимое файла и записываем его на диск
-    # with open(file_location, "wb") as f:
-    #     f.write(file.read())
+
+@app.post("/transcribe/")
+async def transcribe(file: UploadFile = File(...)):
+    # Сохраняем загруженный файл временно
+    file_location = f"temp_{file.filename}"
+
+    # Читаем содержимое файла и записываем его на диск
+    with open(file_location, "wb") as f:
+        f.write(await file.read())
 
     # Детектируем язык
-    language = detect_language_local(file)
+    language = detect_language_local(file_location)
 
     # Обрабатываем файл (транскрипция и сегментация)
-    text, diarized = transcribe_and_diarize_audio(filename=file, language=language)
+    text, diarized = transcribe_and_diarize_audio(filename=file_location, language=language)
 
-    return diarized
+    return {"diarized": diarized}
 
-# Создаем интерфейс с загрузкой файла и функцией обработки
-iface = gr.Interface(
-    fn=transcribe,
-    inputs=gr.Audio(type="filepath"),  # используем тип Audio для загрузки аудиофайлов
-    outputs="text",  # результатом будет текст
-    title="Audio Transcription and Diarization",
-    description="Upload an audio file for transcription and speaker diarization."
-)
 
-# Запускаем приложение
 if __name__ == "__main__":
-    iface.launch()
+    uvicorn.run(app, host="0.0.0.0", port=8000)