Update app.py

app.py

@@ -25,6 +25,13 @@ def generate_text(user_message, system_message, max_tokens, temperature, top_p):
 def upload_excel(file):
     df = pd.read_excel(file.name)
 
+    # 엑셀 파일의 열 이름 확인
+    print("엑셀 파일의 열 이름:", df.columns)
+
+    # D열이 존재하는지 확인하고, 없으면 KeyError 방지
+    if 'D' not in df.columns:
+        return "Error: 엑셀 파일에 'D'열이 존재하지 않습니다."
+
     # 1. G1셀에 "글자수" 입력
     df.loc[0, 'G'] = "글자수"
 
@@ -35,14 +42,20 @@ def upload_excel(file):
     df = df.sort_values(by='G', ascending=False)
 
     # 4. 긍정 리뷰 10개 선택 (리뷰점수 E열이 5점 또는 4점이고, G열의 글자수가 500 이하인 항목)
+    if 'E' not in df.columns:
+        return "Error: 엑셀 파일에 'E'열이 존재하지 않습니다."
+
     positive_reviews = df[(df['E'].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)
 
     # 6. 긍정리뷰와 부정리뷰에서 리뷰날짜, 옵션, 리뷰내용(D열)을 선택하여 반환
-    positive_reviews_data = positive_reviews[['리뷰날짜', '옵션', 'D']]
-    negative_reviews_data = negative_reviews[['리뷰날짜', '옵션', 'D']]
+    if all(col in df.columns for col in ['리뷰날짜', '옵션', 'D']):
+        positive_reviews_data = positive_reviews[['리뷰날짜', '옵션', 'D']]
+        negative_reviews_data = negative_reviews[['리뷰날짜', '옵션', 'D']]
+    else:
+        return "Error: 필요한 열(리뷰날짜, 옵션, D)이 엑셀 파일에 존재하지 않습니다."
 
     # 긍정리뷰와 부정리뷰 내용을 텍스트 형식으로 반환 (그리디오에 보여주기 위해)
     return positive_reviews_data.to_string(), negative_reviews_data.to_string()