Update app.py

app.py

@@ -1,27 +1,58 @@
 import gradio as gr
 import os
 import subprocess
+import shutil
+from huggingface_hub import HfApi, HfFolder, snapshot_download
 
-# Make sure repo folder exists
-if not os.path.exists("storage"):
-    os.makedirs("storage")
+# === Config ===
+DATASET_REPO = "Devity4756/Terminal"  # change to your dataset repo
+HF_TOKEN = HfFolder.get_token()       # will read from your Space secret
+api = HfApi(token=HF_TOKEN)
 
+WORKDIR = "workspace"
+os.makedirs(WORKDIR, exist_ok=True)
+
+# === Restore previous state on startup ===
+try:
+    snapshot_path = snapshot_download(
+        repo_id=DATASET_REPO,
+        repo_type="dataset",
+        token=HF_TOKEN
+    )
+    shutil.copytree(snapshot_path, WORKDIR, dirs_exist_ok=True)
+except Exception as e:
+    print("No previous data restored:", e)
+
+# === Command runner ===
 def run_command(cmd):
     try:
-        # Run inside storage so data persists during Space lifetime
         result = subprocess.run(
-            cmd, shell=True, cwd="storage",
+            cmd, shell=True, cwd=WORKDIR,
             capture_output=True, text=True
         )
+
+        # After running, sync everything in WORKDIR back to dataset
+        try:
+            api.upload_folder(
+                folder_path=WORKDIR,
+                repo_id=DATASET_REPO,
+                repo_type="dataset",
+                commit_message=f"Sync after: {cmd}",
+                token=HF_TOKEN
+            )
+        except Exception as e:
+            print("Upload failed:", e)
+
         return result.stdout + result.stderr
     except Exception as e:
         return str(e)
 
+# === Gradio UI ===
 with gr.Blocks() as demo:
-    gr.Markdown("## 🚀 Hugging Face Mini Terminal with Persistence")
+    gr.Markdown("## 🚀 Hugging Face Terminal (Auto-Sync + Auto-Restore)")
     cmd = gr.Textbox(label="Enter command", placeholder="e.g. git clone https://github.com/user/repo.git")
-    out = gr.Textbox(label="Output")
-    run_btn = gr.Button("Run")
+    out = gr.Textbox(label="Output", lines=15)
+    run_btn = gr.Button("Run Command")
     run_btn.click(run_command, inputs=cmd, outputs=out)
 
 demo.launch()