Update app.py

app.py

@@ -7,6 +7,8 @@ from openpyxl.drawing.image import Image
 from fastapi import FastAPI, UploadFile, File, HTTPException
 from fastapi.responses import FileResponse
 import io
+import tempfile
+import shutil
 
 app = FastAPI()
 
@@ -33,10 +35,14 @@ def extract_keywords(product_names):
 
     # Create the filename with the current date and time
     filename = f'소싱부스트_키워드분석기_{formatted_date}.xlsx'
-    df.to_excel(filename, index=False, startrow=3)  # Save the DataFrame starting from A4
+    
+    # Save the DataFrame to a temporary file
+    temp_dir = tempfile.mkdtemp()
+    file_path = f"{temp_dir}/{filename}"
+    df.to_excel(file_path, index=False, startrow=3)  # Save the DataFrame starting from A4
 
     # Load the workbook and edit the cells
-    wb = load_workbook(filename)
+    wb = load_workbook(file_path)
     ws = wb.active
 
     # Insert the image
@@ -51,8 +57,8 @@ def extract_keywords(product_names):
     ws['D2'].hyperlink = "https://www.ssboost.co.kr"
     ws['D2'].style = "Hyperlink"
 
-    wb.save(filename)
-    return filename
+    wb.save(file_path)
+    return file_path
 
 @app.post("/extract_keywords_from_file/")
 async def extract_keywords_from_file(file: UploadFile = File(...)):
@@ -62,18 +68,20 @@ async def extract_keywords_from_file(file: UploadFile = File(...)):
         if df.empty:
             raise HTTPException(status_code=400, detail="No data found in the specified range.")
         unique_product_names = df.iloc[:, 0].dropna().astype(str).unique().tolist()
-        output_filename = extract_keywords(unique_product_names)
-        return FileResponse(output_filename, media_type='application/vnd.openxmlformats-officedocument.spreadsheetml.sheet', filename=output_filename)
+        file_path = extract_keywords(unique_product_names)
+        return FileResponse(file_path, media_type='application/vnd.openxmlformats-officedocument.spreadsheetml.sheet', filename=file_path.split("/")[-1])
     except Exception as e:
         raise HTTPException(status_code=500, detail=f"An error occurred: {str(e)}")
+    finally:
+        shutil.rmtree(temp_dir)
 
 @app.post("/extract_keywords_from_text/")
 async def extract_keywords_from_text(text: str):
     if not text.strip():
         raise HTTPException(status_code=400, detail="No text provided.")
     product_names = text.split('\n')
-    output_filename = extract_keywords(product_names)
-    return FileResponse(output_filename, media_type='application/vnd.openxmlformats-officedocument.spreadsheetml.sheet', filename=output_filename)
+    file_path = extract_keywords(product_names)
+    return FileResponse(file_path, media_type='application/vnd.openxmlformats-officedocument.spreadsheetml.sheet', filename=file_path.split("/")[-1])
 
 if __name__ == "__main__":
     import uvicorn