Update app.py

app.py

@@ -15,7 +15,7 @@ st.markdown("""
         white-space: nowrap; /* 防止標題換行 */
         overflow: hidden;
         text-overflow: ellipsis;
-        font-size: 2rem; /* 調整字體大小為 2rem，約 32px，根據需要調整 */
+        font-size: 2rem; /* 調整字體大小為 2rem，約 32px */
     }
     .stImage > img {
         max-width: 95%; /* 圖片寬度幾乎填滿 */
@@ -139,18 +139,23 @@ def main():
             # 讀取CSV檔案，強制所有欄位為字串以保留前導零
             df = pd.read_csv(
                 StringIO(uploaded_file.getvalue().decode('utf-8')),
-                header=None,
+                header=0,  # 假設第一行為標題
                 dtype=str  # 所有欄位都以字串形式讀取
             )
             
-            # 自動將第一欄設為姓名，第二欄設為幸運數字
-            df.columns = ['姓名', '幸運號碼', *df.columns[2:]]  # 保留額外欄位（如果有）
+            # 篩選並重新命名需要的欄位，忽略「時間戳記」
+            required_columns = {
+                "您的姓名": "姓名",
+                "你的幸運號碼是？(可重複，前面也可以是0。Ex. 0000013、1111111)": "幸運號碼"
+            }
+            df = df[list(required_columns.keys())].rename(columns=required_columns)
             
-            # 檢查第一行是否為標題（例如 "姓名, 幸運號碼"）
-            if df.iloc[0, 0] == "姓名" and df.iloc[0, 1] == "幸運號碼":
-                df = df.drop(0).reset_index(drop=True)  # 移除標題行
+            # 檢查是否有必要的欄位
+            if not all(col in df.columns for col in ["姓名", "幸運號碼"]):
+                st.error("CSV 檔案缺少 '姓名' 或 '幸運號碼' 欄位，請檢查表單結構。")
+                return
             
-            # 顯示原始資料
+            # 顯示原始資料（僅顯示篩選後的欄位）
             st.subheader("原始資料")
             st.dataframe(df)