Add git branch DAG visualizer with remote branch support

- Replaces placeholder greet app with full branch graph visualizer
- Shows local (green) and remote (blue) branch badges on commits
- Ghost node previews where the next commit will land
- Splits logic into git_graph.py, keeps app.py as pure Gradio UI
- Pins dependency versions for reproducible HF Space builds

Co-Authored-By: Claude Sonnet 4.6 <noreply@anthropic.com>

README.md

@@ -1,15 +1,22 @@
 ---
-title: Git Commit Ai
-emoji: 🦀
+title: Git Branch Visualizer
+emoji: 🌿
 colorFrom: blue
-colorTo: red
+colorTo: green
 sdk: gradio
 sdk_version: 6.17.3
-python_version: '3.13'
+python_version: '3.11'
 app_file: app.py
 pinned: false
 license: mit
-short_description: Generate Automatic Git commit message
+short_description: Visualize git branch history as a DAG with next-commit preview
 ---
 
-Check out the configuration reference at https://huggingface.co/docs/hub/spaces-config-reference
+# Git Branch Graph Visualizer
+
+Visualize your git repository's branch history as an interactive DAG.
+
+- Local branches shown with **green** badges
+- Remote tracking branches shown with **blue** badges
+- **HEAD** commit highlighted with a white ring
+- Type a commit message to preview the **gold ghost node** — shows exactly where the next commit will land

app.py

@@ -3,9 +3,11 @@ import gradio as gr
 from git import InvalidGitRepositoryError
 from git_graph import render_branch_graph
 
+DEFAULT_REPO_PATH = os.environ.get("REPO_PATH", os.getcwd())
+
 
 def draw_graph(repo_path: str, next_commit_message: str):
-    repo_path = repo_path.strip() or os.getcwd()
+    repo_path = repo_path.strip() or DEFAULT_REPO_PATH
     try:
         return render_branch_graph(repo_path, next_commit_message)
     except InvalidGitRepositoryError:
@@ -15,14 +17,14 @@ def draw_graph(repo_path: str, next_commit_message: str):
 with gr.Blocks(title="Git Branch Visualizer", theme=gr.themes.Base()) as demo:
     gr.Markdown("## Git Branch Graph Visualizer")
     gr.Markdown(
-        "Shows all local branches and their commits. "
+        "Shows all local **and remote** branches as a commit DAG. "
         "Type a commit message to preview **where the next commit lands** (gold ghost node)."
     )
 
     with gr.Row():
         repo_path_input = gr.Textbox(
             label="Repository path",
-            value=os.getcwd(),
+            value=DEFAULT_REPO_PATH,
             placeholder="/path/to/your/repo",
             scale=3,
         )

git_graph.py

@@ -41,14 +41,23 @@ def get_commit_graph(repo: Repo) -> list[dict]:
 
 
 def tag_commits_with_branches(repo: Repo, commits: list[dict]) -> list[dict]:
-    sha_to_labels: dict[str, list[str]] = {}
+    sha_to_local: dict[str, list[str]] = {}
     for branch in repo.branches:
         tip = branch.commit.hexsha
-        sha_to_labels.setdefault(tip, []).append(branch.name)
+        sha_to_local.setdefault(tip, []).append(branch.name)
+
+    sha_to_remote: dict[str, list[str]] = {}
+    for remote in repo.remotes:
+        for ref in remote.refs:
+            if ref.name.endswith("/HEAD"):
+                continue
+            tip = ref.commit.hexsha
+            sha_to_remote.setdefault(tip, []).append(ref.name)
 
     head_sha = repo.head.commit.hexsha
     for commit in commits:
-        commit["labels"] = sha_to_labels.get(commit["sha"], [])
+        commit["local_labels"] = sha_to_local.get(commit["sha"], [])
+        commit["remote_labels"] = sha_to_remote.get(commit["sha"], [])
         commit["is_head"] = commit["sha"] == head_sha
     return commits
 
@@ -93,7 +102,8 @@ def inject_ghost_node(commits: list[dict], head_sha: str, current_branch: str, m
         "short_sha": "next",
         "message": message.strip(),
         "parents": [head_sha],
-        "labels": [f"← {current_branch} (next)"],
+        "local_labels": [f"← {current_branch} (next)"],
+        "remote_labels": [],
         "is_head": False,
     }
     return [ghost] + commits
@@ -167,9 +177,8 @@ def build_figure(repo: Repo, commits: list[dict], current_branch: str, next_comm
 
         badge_offset_x = x + 0.35
         badge_offset_y = y + 0.22
-        for label in commit["labels"]:
-            is_current = label == current_branch
-            badge_bg = "#9E6A03" if is_current else "#238636"
+        for label in commit["local_labels"]:
+            badge_bg = "#9E6A03" if label == current_branch else "#238636"
             ax.text(
                 badge_offset_x, badge_offset_y,
                 f"  {label}  ",
@@ -179,6 +188,16 @@ def build_figure(repo: Repo, commits: list[dict], current_branch: str, next_comm
                 zorder=7,
             )
             badge_offset_x += len(label) * 0.075 + 0.6
+        for label in commit["remote_labels"]:
+            ax.text(
+                badge_offset_x, badge_offset_y,
+                f"  {label}  ",
+                va="center", ha="left",
+                fontsize=7, color="white", fontfamily="monospace",
+                bbox=dict(boxstyle="round,pad=0.25", facecolor="#1F6FEB", edgecolor="none"),
+                zorder=7,
+            )
+            badge_offset_x += len(label) * 0.075 + 0.6
 
     ax.set_xlim(-0.8, num_lanes * X_SPACING + 6)
     ax.set_ylim(len(commits) * Y_SPACING - 0.8, 0.8)
@@ -193,11 +212,16 @@ def build_info(repo: Repo, current_branch: str, next_commit_message: str) -> str
     head_sha = repo.head.commit.hexsha
     staged_files = [item.a_path for item in repo.index.diff("HEAD")]
     untracked = repo.untracked_files
+    remote_branches = [
+        ref.name for remote in repo.remotes for ref in remote.refs if not ref.name.endswith("/HEAD")
+    ]
 
     info_lines = [
         f"**Current branch:** `{current_branch}`",
         f"**HEAD:** `{head_sha[:7]}`",
     ]
+    if remote_branches:
+        info_lines.append("**Remote branches:** " + ", ".join(f"`{r}`" for r in remote_branches))
     if staged_files:
         info_lines.append("**Staged (ready to commit):** " + ", ".join(f"`{f}`" for f in staged_files))
     if untracked:

requirements.txt

@@ -1,3 +1,3 @@
-gradio
-gitpython
-matplotlib
+gradio==6.17.3
+gitpython==3.1.50
+matplotlib==3.10.9