Update app.py

app.py

@@ -25,40 +25,40 @@ def generate_text(user_message, system_message, max_tokens, temperature, top_p):
 def upload_excel(file):
     df = pd.read_excel(file.name)
 
-    # 열 이름에 공백이나 특수문자가 포함된 경우 대비하여 열 이름 정리
+    # 열 이름 정리
     df.columns = df.columns.str.strip()
 
     # 열 이름 출력 확인 (디버깅용)
     print("엑셀 파일의 열 이름:", df.columns.tolist())
 
-    # D열이 존재하는지 확인
-    if 'D' not in df.columns:
-        return "Error: 엑셀 파일에 'D'열이 존재하지 않습니다. 열 이름을 확인해주세요."
+    # '리뷰내용' 열을 사용 (기존 'D'열을 대체)
+    if '리뷰내용' not in df.columns:
+        return "Error: 엑셀 파일에 '리뷰내용' 열이 존재하지 않습니다.", ""
 
     # 1. G1셀에 "글자수" 입력
     df.loc[0, 'G'] = "글자수"
 
-    # 2. G2셀부터 G열에 D열의 글자수 입력
-    df['G'] = df['D'].apply(lambda x: len(str(x)) if pd.notnull(x) else 0)
+    # 2. G2셀부터 G열에 '리뷰내용' 열의 글자수 입력
+    df['G'] = df['리뷰내용'].apply(lambda x: len(str(x)) if pd.notnull(x) else 0)
 
     # 3. G열 기준으로 내림차순 정렬
     df = df.sort_values(by='G', ascending=False)
 
     # 4. 긍정 리뷰 10개 선택 (리뷰점수 E열이 5점 또는 4점이고, G열의 글자수가 500 이하인 항목)
-    if 'E' not in df.columns:
-        return "Error: 엑셀 파일에 'E'열이 존재하지 않습니다."
+    if '리뷰점수' not in df.columns:
+        return "Error: 엑셀 파일에 '리뷰점수' 열이 존재하지 않습니다.", ""
 
-    positive_reviews = df[(df['E'].isin([5, 4])) & (df['G'] <= 500)].head(10)
+    positive_reviews = df[(df['리뷰점수'].isin([5, 4])) & (df['G'] <= 500)].head(10)
 
     # 5. 부정 리뷰 10개 선택 (리뷰점수 E열이 1점 또는 2점이고, G열의 글자수가 500 이하인 항목)
-    negative_reviews = df[(df['E'].isin([1, 2])) & (df['G'] <= 500)].head(10)
+    negative_reviews = df[(df['리뷰점수'].isin([1, 2])) & (df['G'] <= 500)].head(10)
 
-    # 6. 긍정리뷰와 부정리뷰에서 리뷰날짜, 옵션, 리뷰내용(D열)을 선택하여 반환
-    if all(col in df.columns for col in ['리뷰날짜', '옵션', 'D']):
-        positive_reviews_data = positive_reviews[['리뷰날짜', '옵션', 'D']]
-        negative_reviews_data = negative_reviews[['리뷰날짜', '옵션', 'D']]
+    # 6. 긍정리뷰와 부정리뷰에서 리뷰날짜, 구매옵션, 리뷰내용을 선택하여 반환
+    if all(col in df.columns for col in ['리뷰날짜', '구매옵션', '리뷰내용']):
+        positive_reviews_data = positive_reviews[['리뷰날짜', '구매옵션', '리뷰내용']]
+        negative_reviews_data = negative_reviews[['리뷰날짜', '구매옵션', '리뷰내용']]
     else:
-        return "Error: 필요한 열(리뷰날짜, 옵션, D)이 엑셀 파일에 존재하지 않습니다."
+        return "Error: 필요한 열(리뷰날짜, 구매옵션, 리뷰내용)이 엑셀 파일에 존재하지 않습니다.", ""
 
     # 긍정리뷰와 부정리뷰 내용을 텍스트 형식으로 반환 (그리디오에 보여주기 위해)
     return positive_reviews_data.to_string(), negative_reviews_data.to_string()