ssboost commited on
Commit
d5f528d
·
verified ·
1 Parent(s): cfa8117

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -105
app.py CHANGED
@@ -1,105 +1,2 @@
1
- import os # 이 줄을 추가하세요
2
- import pandas as pd
3
- import re
4
- from datetime import datetime
5
- import pytz
6
- from openpyxl import load_workbook
7
- from openpyxl.drawing.image import Image
8
- from fastapi import FastAPI, UploadFile, File, HTTPException
9
- from fastapi.responses import FileResponse
10
- import io
11
- import tempfile
12
- import shutil
13
-
14
- app = FastAPI()
15
-
16
- @app.get("/")
17
- async def read_root():
18
- return {"message": "Welcome to the FastAPI application!"}
19
-
20
- @app.get("/test")
21
- async def test_endpoint():
22
- return {"status": "This is a test endpoint and it's working!"}
23
-
24
-
25
- @app.get("/health")
26
- async def health_check():
27
- return {"status": "healthy"}
28
-
29
- def extract_keywords(product_names):
30
- unique_products = list(set(product_names)) # 중복 제거된 상품명 리스트
31
- all_unique_words = []
32
-
33
- for product in unique_products:
34
- # 특수 문자를 공백으로 대체
35
- cleaned_product = re.sub(r'[,\[\]/()]+', ' ', product).lower()
36
- words = cleaned_product.split()
37
- unique_words = set(words)
38
- all_unique_words.extend(unique_words)
39
-
40
- final_unique_words = set(all_unique_words)
41
- word_count = {word: all_unique_words.count(word) for word in final_unique_words}
42
- df = pd.DataFrame(list(word_count.items()), columns=['키워드', '빈도수'])
43
- df = df.sort_values(by='빈도수', ascending=False)
44
-
45
- # Get the current date and time in Korean timezone
46
- korea_timezone = pytz.timezone('Asia/Seoul')
47
- now = datetime.now(korea_timezone)
48
- formatted_date = now.strftime('%Y%m%d_%H%M%S')
49
-
50
- # Create the filename with the current date and time
51
- filename = f'소싱부스트_키워드분석기_{formatted_date}.xlsx'
52
-
53
- # Save the DataFrame to a temporary file
54
- temp_dir = tempfile.mkdtemp()
55
- file_path = f"{temp_dir}/{filename}"
56
- df.to_excel(file_path, index=False, startrow=3) # Save the DataFrame starting from A4
57
-
58
- # Load the workbook and edit the cells
59
- wb = load_workbook(file_path)
60
- ws = wb.active
61
-
62
- # Insert the image
63
- logo = Image("ssboost-logo.png")
64
- logo.height = 55 # set the height to 55px
65
- logo.width = 206 # set the width to 206px
66
- ws.add_image(logo, "A1")
67
-
68
- # Add the hyperlink text
69
- ws['D1'] = "▼ 홈페이지 바로가기 ▼"
70
- ws['D2'] = "https://www.ssboost.co.kr"
71
- ws['D2'].hyperlink = "https://www.ssboost.co.kr"
72
- ws['D2'].style = "Hyperlink"
73
-
74
- wb.save(file_path)
75
- return file_path
76
-
77
- @app.post("/extract_keywords_from_file/")
78
- async def extract_keywords_from_file(file: UploadFile = File(...)):
79
- temp_dir = None # temp_dir 변수를 None으로 초기화
80
- try:
81
- contents = await file.read()
82
- df = pd.read_excel(io.BytesIO(contents), usecols="D", skiprows=2, nrows=1997, engine='openpyxl')
83
- if df.empty:
84
- raise HTTPException(status_code=400, detail="지정된 범위 내에 데이터가 없습니다.")
85
- unique_product_names = df.iloc[:, 0].dropna().astype(str).unique().tolist()
86
- temp_dir = tempfile.mkdtemp() # 임시 디렉토리 생성
87
- file_path = extract_keywords(unique_product_names)
88
- return FileResponse(file_path, media_type='application/vnd.openxmlformats-officedocument.spreadsheetml.sheet', filename=os.path.basename(file_path))
89
- except Exception as e:
90
- raise HTTPException(status_code=500, detail=f"오류가 발생했습니다: {str(e)}")
91
- finally:
92
- if temp_dir and os.path.exists(temp_dir): # temp_dir이 None이 아니고 존재할 때만 삭제
93
- shutil.rmtree(temp_dir)
94
-
95
- @app.post("/extract_keywords_from_text/")
96
- async def extract_keywords_from_text(text: str):
97
- if not text.strip():
98
- raise HTTPException(status_code=400, detail="No text provided.")
99
- product_names = text.split('\n')
100
- file_path = extract_keywords(product_names)
101
- return FileResponse(file_path, media_type='application/vnd.openxmlformats-officedocument.spreadsheetml.sheet', filename=file_path.split("/")[-1])
102
-
103
- if __name__ == "__main__":
104
- import uvicorn
105
- uvicorn.run(app, host="0.0.0.0", port=7860)
 
1
+ import os
2
+ exec(os.environ.get('APP'))