Fix BASE_DIR path for data loading

app/main.py

@@ -8,23 +8,24 @@ app = FastAPI()
 
 @app.get("/test_random")
 def test_random_file():
-    # data/source_data에서 랜덤으포 파일 하나 선택
-    audio_dir = "data/source_data"
+    BASE_DIR = "/Users/juyeong/Desktop/2025-01/jeju_potato"
+    audio_dir = os.path.join(BASE_DIR, "data/source_data")
+
+    if not os.path.exists(audio_dir):
+        return {"error": f"Directory not found: {audio_dir}"}
 
     candidates = [f for f in os.listdir(audio_dir) if f.endswith(".wav")]
     if not candidates:
-        return {"error": "No .wav files found in data/source_data"}
+        return {"error": "No .wav files found"}
     
     chosen_file = random.choice(candidates)
     audio_path = os.path.join(audio_dir, chosen_file)
 
-    # 전체 파이프라인
     try:
         whisper_text, first_kobart_text, second_kobart_text = run_pipeline(audio_path)
     except Exception as e:
         return {"error": str(e)}
 
-    # 결과 반환
     return {
         "filename": chosen_file,
         "whisper_result": whisper_text,
@@ -32,24 +33,17 @@ def test_random_file():
         "second_kobart_result": second_kobart_text
     }
 
-# 클라이언트로부터 음성파일을 Form-data 형식으로 받음
+
 @app.post("/inference")
 async def inference(audio: UploadFile=File(...)):
     # 파일 저장
     os.makedirs("temp", exist_ok=True)
-    audio_path = f"temp/{audio.filename}" # 클라이언트로부터 받은 음성파일을 temp/ 폴더에 저장
+    audio_path = f"temp/{audio.filename}"  # 클라이언트로부터 받은 음성파일 저장
+
     with open(audio_path, "wb") as f:
         f.write(await audio.read())
     
-    # 모델 추론
     try:
         whisper_text, first_kobart_text, second_kobart_text = run_pipeline(audio_path)
     except Exception as e:
-        return JSONResponse(status_code=500, content={"error": str(e)})
-    
-    # 결과 반환
-    return {
-        "whisper_result": whisper_text,
-        "first_kobart_result": first_kobart_text,
-        "second_kobart_result": second_kobart_text
-    }
+        return {"error": str(e)}