Update app.py

app.py

@@ -23,6 +23,13 @@ def process_number(number):
 def main():
     st.title("幸運數字查詢系統")
     
+    # 讓使用者選擇幸運數字的位數
+    selected_digits = st.selectbox(
+        "請選擇幸運數字的位數",
+        options=range(4, 11),  # 4 到 10 位
+        index=3  # 預設選 7 位
+    )
+    
     # 檔案上傳部分
     uploaded_file = st.file_uploader("請上傳CSV檔案", type=['csv'])
     
@@ -64,11 +71,20 @@ def main():
                 progress_bar.progress(progress)
                 status_text.text(f"正在處理: {index + 1}/{total_rows}")
                 
-                # 獲取幸運數字並確保為7位數
-                lucky_number = str(row['幸運號碼']).zfill(7)  # 小於7位時補零
-                result = process_number(lucky_number)
-                df.at[index, '幸運號碼'] = lucky_number  # 更新為補零後的數字
-                df.at[index, '查詢結果'] = result
+                # 獲取幸運數字並檢查位數
+                original_number = str(row['幸運號碼'])
+                number_length = len(original_number)
+                
+                if number_length > selected_digits:
+                    # 超過指定位數，標記為不符規定
+                    df.at[index, '幸運號碼'] = original_number  # 保留原始輸入
+                    df.at[index, '查詢結果'] = "數字不符規定"
+                else:
+                    # 小於或等於指定位數，補零後查詢
+                    lucky_number = original_number.zfill(selected_digits)
+                    result = process_number(lucky_number)
+                    df.at[index, '幸運號碼'] = lucky_number  # 更新為補零後的數字
+                    df.at[index, '查詢結果'] = result
             
             # 顯示結果
             st.subheader("完整結果")