Update data_app.py

data_app.py

@@ -7,24 +7,34 @@ from huggingface_hub import HfApi
 st.set_page_config(page_title="🔧 Service Tech Tracker", layout="wide")
 st.title("🔧 Service Tech Fuel & Job Tracker")
 
-# ====================== HF TOKEN SETUP ======================
-HF_TOKEN = st.secrets.get("HF_TOKEN")
-if not HF_TOKEN:
-    HF_TOKEN = st.sidebar.text_input("Enter your HF Token (for backup)", type="password")
+# ====================== HF TOKEN (Safe Version) ======================
+HF_TOKEN = None
+if "HF_TOKEN" in st.secrets:
+    HF_TOKEN = st.secrets["HF_TOKEN"]
+else:
+    HF_TOKEN = st.sidebar.text_input("🔑 Hugging Face Token (for backup)", type="password", key="hf_input")
 
 HF_REPO = "AddisonSwan/service-tech-tracker"
 
-def backup_to_hf(filename="mileage.csv"):
-    if HF_TOKEN and os.path.exists(filename):
+def backup_to_hf():
+    if not HF_TOKEN:
+        return
+    if os.path.exists("data/mileage.csv"):
         try:
             api = HfApi(token=HF_TOKEN)
-            api.upload_file(path_or_fileobj=filename, path_in_repo=filename, repo_id=HF_REPO, repo_type="dataset")
-            st.success("✅ Backed up to Hugging Face")
+            api.upload_file(
+                path_or_fileobj="data/mileage.csv",
+                path_in_repo="mileage.csv",
+                repo_id=HF_REPO,
+                repo_type="dataset"
+            )
+            st.sidebar.success("✅ Backed up to HF")
         except:
-            st.warning("Backup failed - check token")
+            st.sidebar.warning("Backup failed")
 
 # Create folders
 os.makedirs("data", exist_ok=True)
+os.makedirs("jobs", exist_ok=True)
 
 page = st.sidebar.radio("Go to", ["Gas & Mileage", "New Job", "All Data"])
 
@@ -57,9 +67,9 @@ if page == "Gas & Mileage":
             df = pd.concat([df, new_row], ignore_index=True)
         except:
             df = new_row
+            
         df.to_csv("data/mileage.csv", index=False)
-        
-        backup_to_hf("data/mileage.csv")
+        backup_to_hf()
         st.success("✅ Saved!")
 
     if os.path.exists("data/mileage.csv"):
@@ -72,11 +82,11 @@ elif page == "New Job":
     desc = st.text_area("Description")
     if st.button("Create Job"):
         os.makedirs(f"jobs/{job_id}", exist_ok=True)
-        st.success(f"Job {job_id} created!")
+        st.success(f"✅ Job {job_id} created!")
 
 elif page == "All Data":
     st.subheader("All Records")
     if os.path.exists("data/mileage.csv"):
         st.dataframe(pd.read_csv("data/mileage.csv"))
 
-st.caption("Service Tech Tracker • Connected to HF")
+st.caption("Service Tech Tracker • Running on Hugging Face")