Update app.py

app.py

@@ -29,53 +29,54 @@ def main():
     if uploaded_file:
         try:
             # 讀取CSV檔案
-            df = pd.read_csv(StringIO(uploaded_file.getvalue().decode('utf-8')))
+            df = pd.read_csv(StringIO(uploaded_file.getvalue().decode('utf-8')), header=None)
+            
+            # 自動將第一欄設為姓名，第二欄設為幸運數字
+            df.columns = ['姓名', '幸運號碼', *df.columns[2:]]  # 保留額外欄位（如果有）
+            
+            # 檢查第一行是否為標題（例如 "姓名, 幸運號碼"）
+            if df.iloc[0, 0] == "姓名" and df.iloc[0, 1] == "幸運號碼":
+                df = df.drop(0).reset_index(drop=True)  # 移除標題行
             
             # 顯示原始資料
             st.subheader("原始資料")
             st.dataframe(df)
             
-            # 確認欄位名稱存在
-            lucky_number_column = "你的幸運號碼是？(可重複，前面也可以是0。Ex. 0000013、1111111)"
-            if lucky_number_column in df.columns:
-                # 處理每個幸運數字
-                st.subheader("處理結果")
-                
-                # 建立進度條
-                progress_bar = st.progress(0)
-                status_text = st.empty()
-                
-                # 新增結果欄位
-                df['查詢結果'] = None
-                
-                # 處理每個數字
-                total_rows = len(df)
-                for index, row in df.iterrows():
-                    # 更新進度
-                    progress = (index + 1) / total_rows
-                    progress_bar.progress(progress)
-                    status_text.text(f"正在處理: {index + 1}/{total_rows}")
-                    
-                    # 獲取並處理幸運數字
-                    lucky_number = str(row[lucky_number_column])
-                    result = process_number(lucky_number)
-                    df.at[index, '查詢結果'] = result
-                
-                # 顯示結果
-                st.subheader("完整結果")
-                st.dataframe(df)
-                
-                # 提供下載功能
-                csv = df.to_csv(index=False)
-                st.download_button(
-                    label="下載處理結果",
-                    data=csv,
-                    file_name="processed_results.csv",
-                    mime="text/csv"
-                )
+            # 處理每個幸運數字
+            st.subheader("處理結果")
+            
+            # 建立進度條
+            progress_bar = st.progress(0)
+            status_text = st.empty()
+            
+            # 新增結果欄位
+            df['查詢結果'] = None
+            
+            # 處理每個數字
+            total_rows = len(df)
+            for index, row in df.iterrows():
+                # 更新進度
+                progress = (index + 1) / total_rows
+                progress_bar.progress(progress)
+                status_text.text(f"正在處理: {index + 1}/{total_rows}")
                 
-            else:
-                st.error("找不到幸運數字欄位，請確認CSV檔案格式是否正確")
+                # 獲取並處理幸運數字
+                lucky_number = str(row['幸運號碼'])
+                result = process_number(lucky_number)
+                df.at[index, '查詢結果'] = result
+            
+            # 顯示結果
+            st.subheader("完整結果")
+            st.dataframe(df)
+            
+            # 提供下載功能
+            csv = df.to_csv(index=False)
+            st.download_button(
+                label="下載處理結果",
+                data=csv,
+                file_name="processed_results.csv",
+                mime="text/csv"
+            )
                 
         except Exception as e:
             st.error(f"處理檔案時發生錯誤: {str(e)}")