Update app.py

app.py

@@ -204,26 +204,40 @@ def main():
     
     if uploaded_file:
         try:
-            # 讀取CSV檔案，強制所有欄位為字串以保留前導零
+            # 嘗試讀取 CSV，假設有表頭
             df = pd.read_csv(
                 StringIO(uploaded_file.getvalue().decode('utf-8')),
-                header=0,  # 假設第一行為標題
+                header=0,  # 預設有表頭
                 dtype=str  # 所有欄位都以字串形式讀取
             )
             
-            # 篩選並重新命名需要的欄位，忽略「時間戳記」
+            # 定義可能的欄位映射
             required_columns = {
                 "您的姓名": "姓名",
-                "你的幸運號碼是？(可重複，前面也可以是0。Ex. 0000013、1111111)": "幸運號碼"
+                "你的幸運號碼是？(可重複，前面也可以是0。Ex. 0000013、1111111)": "幸運號碼",
+                "姓名": "姓名",
+                "自選號碼": "幸運號碼"
             }
-            df = df[list(required_columns.keys())].rename(columns=required_columns)
             
             # 檢查是否有必要的欄位
-            if not all(col in df.columns for col in ["姓名", "幸運號碼"]):
-                st.error("CSV 檔案缺少 '姓名' 或 '幸運號碼' 欄位，請檢查表單結構。")
-                return
+            available_columns = [col for col in required_columns.keys() if col in df.columns]
+            if not available_columns or len(available_columns) < 2:
+                # 若無表頭或缺少必要欄位，假設無表頭，第一列為姓名，第二列為幸運號碼
+                df = pd.read_csv(
+                    StringIO(uploaded_file.getvalue().decode('utf-8')),
+                    header=None,
+                    dtype=str
+                )
+                if len(df.columns) >= 2:
+                    df.columns = ["姓名", "幸運號碼"]
+                else:
+                    st.error("CSV 檔案格式錯誤，缺少 '姓名' 或 '幸運號碼' 欄位。")
+                    return
+            
+            # 篩選並重新命名需要的欄位，忽略「時間戳記」
+            df = df[[col for col in df.columns if col in required_columns]].rename(columns=required_columns)
             
-            # 顯示原始資料（僅顯示篩選後的欄位）
+            # 顯示原始資料
             st.subheader("原始資料")
             st.dataframe(df)