Deploy: copy app, Dockerfile, requirements, and assist to Space

Dockerfile

@@ -0,0 +1,16 @@
+# Use a small Python base image
+FROM python:3.12-slim
+
+WORKDIR /app
+
+# Copy and install dependencies
+COPY requirements.txt .
+RUN pip install --upgrade pip && \
+    pip install -r requirements.txt
+
+# Copy all project files
+COPY . .
+
+# Expose Streamlit port and run
+EXPOSE 8501
+CMD ["streamlit", "run", "app.py", "--server.port=8501", "--server.address=0.0.0.0"]

app.py

@@ -0,0 +1,13 @@
+import streamlit as st
+from assist.chat import chat as chat_plugin
+
+st.title("RepoSage Chatbot Demo")
+
+# 1) Change the label to make it obvious we're asking a question
+question = st.text_input("Ask RepoSage a question:", "")
+
+# 2) Only run when clicked
+if st.button("Ask RepoSage"):
+    # 3) Pass that question into your stub
+    response = chat_plugin(question)
+    st.write(response)

assist/__init__.py

@@ -0,0 +1 @@
+

assist/chat.py

@@ -0,0 +1,9 @@
+# assist/chat.py
+
+def chat(question: str) -> str:
+    """
+    Chat plugin stub: echoes back what you asked.
+    """
+    if not question.strip():
+        return "Please enter a question above."
+    return f"Chat plugin stub received: “{question}”"

assist/heatmap.py

@@ -0,0 +1,14 @@
+import torch
+import matplotlib.pyplot as plt
+from sklearn.metrics.pairwise import cosine_similarity
+
+def show_heatmap(tensor_path="tensor.pt"):
+    # Load embeddings
+    weights = torch.load(tensor_path).numpy()
+    # Compute similarity
+    sim = cosine_similarity(weights)
+    # Plot
+    plt.imshow(sim, cmap="viridis")
+    plt.colorbar()
+    plt.title("Token Similarity Heatmap")
+    plt.show()

assist/main.py

@@ -0,0 +1,29 @@
+import typer
+from rich import print
+from pathlib import Path
+
+# Import and alias the stub so it doesn't collide with our CLI function
+from .chat import chat as chat_plugin
+from .heatmap import show_heatmap
+
+app = typer.Typer(help="RepoSage Synth CLI")
+
+@app.command()
+def hello(name: str = "world"):
+    """Warm-up command: prints Hello, {name}!"""
+    print(f"[bold green]Hello, {name}![/]")
+
+@app.command()
+def heatmap():
+    """Show token-similarity heatmap from tensor.pt."""
+    show_heatmap()
+
+@app.command()
+def chat(repo: Path = Path(".")):
+    """Invoke the chat plugin stub."""
+    # Call the aliased stub, not this function
+    response = chat_plugin(repo)
+    print(response)
+
+if __name__ == "__main__":
+    app()

assist/requirements.txt

@@ -0,0 +1,43 @@
+certifi==2025.6.15
+charset-normalizer==3.4.2
+click==8.2.1
+contourpy==1.3.2
+cycler==0.12.1
+filelock==3.18.0
+fonttools==4.58.4
+fsspec==2025.5.1
+gitdb==4.0.12
+GitPython==3.1.44
+idna==3.10
+Jinja2==3.1.6
+joblib==1.5.1
+kiwisolver==1.4.8
+markdown-it-py==3.0.0
+MarkupSafe==3.0.2
+matplotlib==3.10.3
+mdurl==0.1.2
+mpmath==1.3.0
+networkx==3.5
+nltk==3.9.1
+numpy==2.3.0
+packaging==25.0
+pillow==11.2.1
+Pygments==2.19.1
+pyparsing==3.2.3
+python-dateutil==2.9.0.post0
+regex==2024.11.6
+requests==2.32.4
+rich==14.0.0
+scikit-learn==1.7.0
+scipy==1.15.3
+setuptools==80.9.0
+shellingham==1.5.4
+six==1.17.0
+smmap==5.0.2
+sympy==1.14.0
+threadpoolctl==3.6.0
+torch==2.7.1
+tqdm==4.67.1
+typer==0.16.0
+typing_extensions==4.14.0
+urllib3==2.4.0