Update app.py

app.py

@@ -75,18 +75,21 @@ def extract_keywords(product_names):
 
 @app.post("/extract_keywords_from_file/")
 async def extract_keywords_from_file(file: UploadFile = File(...)):
+    temp_dir = None  # temp_dir 변수를 None으로 초기화
     try:
         contents = await file.read()
         df = pd.read_excel(io.BytesIO(contents), usecols="D", skiprows=2, nrows=1997, engine='openpyxl')
         if df.empty:
-            raise HTTPException(status_code=400, detail="No data found in the specified range.")
+            raise HTTPException(status_code=400, detail="지정된 범위 내에 데이터가 없습니다.")
         unique_product_names = df.iloc[:, 0].dropna().astype(str).unique().tolist()
+        temp_dir = tempfile.mkdtemp()  # 임시 디렉토리 생성
         file_path = extract_keywords(unique_product_names)
-        return FileResponse(file_path, media_type='application/vnd.openxmlformats-officedocument.spreadsheetml.sheet', filename=file_path.split("/")[-1])
+        return FileResponse(file_path, media_type='application/vnd.openxmlformats-officedocument.spreadsheetml.sheet', filename=os.path.basename(file_path))
     except Exception as e:
-        raise HTTPException(status_code=500, detail=f"An error occurred: {str(e)}")
+        raise HTTPException(status_code=500, detail=f"오류가 발생했습니다: {str(e)}")
     finally:
-        shutil.rmtree(temp_dir)
+        if temp_dir and os.path.exists(temp_dir):  # temp_dir이 None이 아니고 존재할 때만 삭제
+            shutil.rmtree(temp_dir)
 
 @app.post("/extract_keywords_from_text/")
 async def extract_keywords_from_text(text: str):