Update app.py

app.py

@@ -1,61 +1,79 @@
 import gradio as gr
 import os
 from datetime import datetime
-from huggingface_hub import HfApi
+from huggingface_hub import HfApi, hf_hub_download, HfHubHTTPError
 
-# ----- CONFIG -----
-# Make sure this matches your Space URL exactly:
-# https://huggingface.co/spaces/aparke21/simple_gradio_app
-REPO_ID = "aparke21/simple_gradio_app"   # <username>/<space-name>
-REPO_TYPE = "space"
-LOG_LOCAL_DIR = "temp_logs"
-LOG_LOCAL_PATH = os.path.join(LOG_LOCAL_DIR, "logs.txt")
-LOG_REMOTE_PATH = "internal_logs/logs.txt"  # where it will live in the Space repo
+# ===== CONFIG =====
+LOG_REPO_ID = "aparke21/simple_gradio_app_logs"  # <-- your dataset repo
+LOG_REPO_TYPE = "dataset"
+LOG_FILE_NAME = "logs.txt"
 
 
-def _append_to_local_log(message: str) -> None:
-    """Append a log line to a local file inside the container."""
-    os.makedirs(LOG_LOCAL_DIR, exist_ok=True)
-    timestamp = datetime.utcnow().isoformat()
-    line = f"{timestamp} | {message}\n"
-    with open(LOG_LOCAL_PATH, "a") as f:
-        f.write(line)
+def _get_token():
+    token = os.getenv("HF_TOKEN")
+    if token is None:
+        print("[LOGGING] HF_TOKEN not set; skipping logging.")
+    return token
 
 
-def _upload_log_to_repo() -> None:
-    """Upload the local log file to the Space repo (overwriting logs.txt)."""
-    token = os.getenv("HF_TOKEN")
+def _download_existing_log(token):
+    """Try to download existing logs.txt from the dataset repo. 
+    Returns local path, or a new empty file path."""
+    local_path = "logs_buffer.txt"
+
+    try:
+        cached_path = hf_hub_download(
+            repo_id=LOG_REPO_ID,
+            repo_type=LOG_REPO_TYPE,
+            filename=LOG_FILE_NAME,
+            token=token,
+        )
+        # Copy cached file to our working file
+        with open(cached_path, "r") as src, open(local_path, "w") as dst:
+            dst.write(src.read())
+        print("[LOGGING] Existing log downloaded and loaded.")
+    except HfHubHTTPError as e:
+        # Likely file does not exist yet (404)
+        print("[LOGGING] No existing log found, starting a new one.", e)
+        # Ensure empty file
+        with open(local_path, "w") as f:
+            f.write("")
+
+    return local_path
+
+
+def _append_and_upload_log(message: str):
+    token = _get_token()
     if token is None:
-        # If the token isn't set, just skip uploading but don't break the app.
-        print("[LOGGING] HF_TOKEN not set; skipping upload to repo.")
-        return
+        return  # no token, cannot log
+
+    api = HfApi(token=token)
 
-    if not os.path.exists(LOG_LOCAL_PATH):
-        print("[LOGGING] Local log file does not exist; nothing to upload.")
-        return
+    # 1. Get current logs from the dataset (if any)
+    local_path = _download_existing_log(token)
+
+    # 2. Append new log line
+    timestamp = datetime.utcnow().isoformat()
+    line = f"{timestamp} | {message}\n"
+    with open(local_path, "a") as f:
+        f.write(line)
 
+    # 3. Upload updated log file back to dataset repo
     try:
-        api = HfApi(token=token)
         api.upload_file(
-            path_or_fileobj=LOG_LOCAL_PATH,
-            repo_id=REPO_ID,
-            repo_type=REPO_TYPE,
-            path_in_repo=LOG_REMOTE_PATH,
+            path_or_fileobj=local_path,
+            repo_id=LOG_REPO_ID,
+            repo_type=LOG_REPO_TYPE,
+            path_in_repo=LOG_FILE_NAME,
         )
-        print(f"[LOGGING] Uploaded log file to {REPO_ID}/{LOG_REMOTE_PATH}")
+        print(f"[LOGGING] Uploaded updated logs to {LOG_REPO_ID}/{LOG_FILE_NAME}")
     except Exception as e:
-        # Don't crash the app if upload fails; just log the error.
-        print("[LOGGING] Failed to upload log file:", e)
+        print("[LOGGING] Failed to upload logs:", e)
 
 
 def greet(name: str) -> str:
-    # 1. Log this interaction locally
-    _append_to_local_log(f"Greet called with name='{name}'")
-
-    # 2. Sync the log to your Space repo (so you can see it later in Files & Versions)
-    _upload_log_to_repo()
-
-    # 3. Normal app behavior
+    # Log the interaction (quietly, server-side)
+    _append_and_upload_log(f"greet called with name='{name}'")
     return "Hello " + name + "!!"
 
 
@@ -63,3 +81,4 @@ demo = gr.Interface(fn=greet, inputs="text", outputs="text")
 
 if __name__ == "__main__":
     demo.launch()
+