Update app.py

app.py

@@ -25,12 +25,15 @@ 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)
+    # 열 이름에 공백이나 특수문자가 포함된 경우 대비하여 열 이름 정리
+    df.columns = df.columns.str.strip()
 
-    # D열이 존재하는지 확인하고, 없으면 KeyError 방지
+    # 열 이름 출력 확인 (디버깅용)
+    print("엑셀 파일의 열 이름:", df.columns.tolist())
+
+    # D열이 존재하는지 확인
     if 'D' not in df.columns:
-        return "Error: 엑셀 파일에 'D'열이 존재하지 않습니다."
+        return "Error: 엑셀 파일에 'D'열이 존재하지 않습니다. 열 이름을 확인해주세요."
 
     # 1. G1셀에 "글자수" 입력
     df.loc[0, 'G'] = "글자수"