From c104ea52b532aa210d6c0249d85ca29f3172e583 Mon Sep 17 00:00:00 2001
From: superobk <superobk@hotmail.com>
Date: Tue, 9 Apr 2024 12:00:08 +0800
Subject: [PATCH] code updated

---
 main.py                    | 10 ++++++++++
 src/blackbox/asr.py        |  2 +-
 src/blackbox/audio_chat.py |  2 +-
 3 files changed, 12 insertions(+), 2 deletions(-)

diff --git a/main.py b/main.py
index 94bbe68..2cb5207 100644
--- a/main.py
+++ b/main.py
@@ -8,9 +8,19 @@ from src.dotchain.runtime.tokenizer import Tokenizer
 from src.dotchain.runtime.runtime import Runtime
 from src.blackbox.blackbox_factory import BlackboxFactory
 import uvicorn
+from fastapi.middleware.cors import CORSMiddleware
 
 app = FastAPI()
 
+app.add_middleware(
+    CORSMiddleware,
+    allow_origins=["*"],
+    allow_credentials=True,
+    allow_methods=["*"],
+    allow_headers=["*"],
+)
+
+
 blackbox_factory = BlackboxFactory()
 
 @app.post("/")
diff --git a/src/blackbox/asr.py b/src/blackbox/asr.py
index 7ba88fb..9c691f3 100644
--- a/src/blackbox/asr.py
+++ b/src/blackbox/asr.py
@@ -38,4 +38,4 @@ class ASR(Blackbox):
             txt = await self.processing(d)
         except ValueError as e:
             return JSONResponse(content={"error": str(e)}, status_code=status.HTTP_400_BAD_REQUEST)
-        return JSONResponse(content={"txt": txt}, status_code=status.HTTP_200_OK)
\ No newline at end of file
+        return JSONResponse(content={"text": txt}, status_code=status.HTTP_200_OK)
\ No newline at end of file
diff --git a/src/blackbox/audio_chat.py b/src/blackbox/audio_chat.py
index 6528a7f..1ab156b 100644
--- a/src/blackbox/audio_chat.py
+++ b/src/blackbox/audio_chat.py
@@ -33,4 +33,4 @@ class AudioChat(Blackbox):
             return JSONResponse(content={"error": "data is required"}, status_code=status.HTTP_400_BAD_REQUEST)
         d = await data.read()
         by = await self.processing(d)
-        return Response(content=by.read(), media_type="audio/wav", headers={"Content-Disposition": "attachment; filename=audio.wav"})
\ No newline at end of file
+        return Response(content=by.read(), media_type="audio/x-wav", headers={"Content-Disposition": "attachment; filename=audio.wav"})
\ No newline at end of file