Upload 18 files

app.py

@@ -157,6 +157,52 @@ def ui_load_model(name):
 _SELFTEST = {"done": False, "result": ""}
 
 
+def _publish_traces(repo_id: str):
+    """One-click: upload /data/traces to the Hub as a dataset, using the
+    HF_TOKEN secret. No command line needed."""
+    repo_id = (repo_id or "").strip()
+    if "/" not in repo_id:
+        return "⚠️ Enter a repo id like `username/dataset-name`."
+    token = (os.environ.get("HF_TOKEN") or os.environ.get("HUGGING_FACE_HUB_TOKEN")
+             or "").strip()
+    if not token:
+        return ("⚠️ No `HF_TOKEN` secret found. Add it in Space → Settings → "
+                "Variables and secrets (a **write** token), then try again.")
+    try:
+        import paths
+        traces = [f for f in os.listdir(paths.TRACES_DIR) if f.endswith(".json")]
+    except OSError:
+        traces = []
+    if not traces:
+        return "⚠️ No traces yet — run a few Auto Research reports first."
+    try:
+        from huggingface_hub import HfApi
+        api = HfApi(token=token)
+        api.create_repo(repo_id, repo_type="dataset", exist_ok=True)
+        # a small README so the dataset page explains itself
+        card = (
+            "---\nlicense: mit\ntags: [agent-trace, finance, chan-theory, llama-cpp]\n---\n\n"
+            "# Chan Compass — agent traces\n\n"
+            "JSON traces from the Chan Compass multi-agent research desk. Each file "
+            "is one ticker's run: the plan, every evidence-tool call and its result, "
+            "and each local sub-agent's request and response. Shared for the "
+            "Build Small hackathon (*Sharing is Caring*).\n\n"
+            "*Educational data — not investment advice.*\n")
+        import tempfile
+        rp = os.path.join(tempfile.gettempdir(), "README.md")
+        with open(rp, "w", encoding="utf-8") as f:
+            f.write(card)
+        api.upload_file(path_or_fileobj=rp, path_in_repo="README.md",
+                        repo_id=repo_id, repo_type="dataset")
+        api.upload_folder(folder_path=paths.TRACES_DIR, path_in_repo="traces",
+                          repo_id=repo_id, repo_type="dataset")
+        url = f"https://huggingface.co/datasets/{repo_id}"
+        return (f"✅ Published **{len(traces)}** trace(s) to [{repo_id}]({url}). "
+                f"Put that link in your submission for the *Sharing is Caring* badge.")
+    except Exception as e:
+        return f"❌ Upload failed: {e}"
+
+
 def _export_dataset():
     n = finetune_data.count()
     if n == 0:
@@ -327,6 +373,17 @@ with gr.Blocks(title="Chan Compass · US", **_style_kw) as demo:
         auto_log = gr.Textbox(lines=14, label="Pipeline log (live — updates every 2s)",
                               elem_id="detail-log")
         traces_md = gr.Markdown(research_agent.list_traces())
+        gr.Markdown("**📡 Share traces** — publish every JSON agent trace in "
+                    "`/data/traces` as a Hugging Face **dataset** (one click, uses "
+                    "your `HF_TOKEN` secret — no command line). Earns *Sharing is "
+                    "Caring*.", elem_classes=["s2-footnote"])
+        with gr.Row():
+            trace_repo = gr.Textbox(
+                value="ranranrunforit/chan-compass-agent-traces",
+                label="Dataset repo id", scale=3)
+            trace_pub = gr.Button("📡 Publish traces as dataset", scale=1)
+        trace_pub_status = gr.Markdown()
+        trace_pub.click(_publish_traces, trace_repo, trace_pub_status)
         auto_log_timer = gr.Timer(2.0)
 
         def _auto_tick():