init commit

.gitattributes

@@ -0,0 +1 @@
+* text=auto eol=lf

.github/workflows/sync_huggingface.yml

@@ -0,0 +1,23 @@
+name: Sync to Hugging Face hub
+on:
+  push:
+    branches: [master]
+
+  # to run this workflow manually from the Actions tab
+  workflow_dispatch:
+
+jobs:
+  sync-to-hub:
+    runs-on: ubuntu-latest
+    steps:
+      - uses: actions/checkout@v3
+        with:
+          fetch-depth: 0
+          lfs: true
+      - name: Checkout main
+        run: git checkout -b main
+      - name: Push to hub
+        env:
+          HF_TOKEN: ${{ secrets.HF_TOKEN }}
+          REPO_ID: codedog-ai/codedog-demo
+        run: git push https://HF_USERNAME:$HF_TOKEN@huggingface.co/spaces/$REPO_ID main

.gitignore

@@ -0,0 +1,143 @@
+# Byte-compiled / optimized / DLL files
+__pycache__/
+*.py[cod]
+*$py.class
+
+# C extensions
+*.so
+
+# Distribution / packaging
+.Python
+build/
+develop-eggs/
+dist/
+downloads/
+eggs/
+.eggs/
+lib/
+lib64/
+parts/
+sdist/
+var/
+wheels/
+pip-wheel-metadata/
+share/python-wheels/
+*.egg-info/
+.installed.cfg
+*.egg
+MANIFEST
+
+# .vscode files
+.vscode/
+# Pycharm
+.idea/
+
+# PyInstaller
+#  Usually these files are written by a python script from a template
+#  before PyInstaller builds the exe, so as to inject date/other infos into it.
+*.manifest
+*.spec
+
+# Installer logs
+pip-log.txt
+pip-delete-this-directory.txt
+
+# Unit test / coverage rerts
+htmlcov/
+.tox/
+.nox/
+.coverage
+.coverage.*
+.cache
+nosetests.xml
+coverage.xml
+*.cover
+*.py,cover
+.hypothesis/
+.pytest_cache/
+
+# Translations
+*.mo
+*.pot
+
+# Django stuff:
+*.log
+local_settings.py
+db.sqlite3
+db.sqlite3-journal
+
+# Flask stuff:
+instance/
+.webassets-cache
+
+# Scrapy stuff:
+.scrapy
+
+# Sphinx documentation
+docs/_build/
+
+# PyBuilder
+target/
+
+# Jupyter Notebook
+.ipynb_checkpoints
+
+# IPython
+profile_default/
+ipython_config.py
+
+# pyenv
+.python-version
+
+# pipenv
+#   According to pypa/pipenv#598, it is recommended to include Pipfile.lock in version control.
+#   However, in case of collaboration, if having platform-specific dependencies or dependencies
+#   having no cross-platform support, pipenv may install dependencies that don't work, or not
+#   install all needed dependencies.
+#Pipfile.lock
+
+# PEP 582; used by e.g. github.com/David-OConnor/pyflow
+__pypackages__/
+
+# Celery stuff
+celerybeat-schedule
+celerybeat.pid
+
+# SageMath parsed files
+*.sage.py
+
+# Environments
+.env
+.env.bat
+.venv
+env/
+venv/
+ENV/
+env.bak/
+venv.bak/
+myvenv/
+tmp/
+*.pem
+
+# Spyder project settings
+.spyderproject
+.spyproject
+
+# Rope project settings
+.ropeproject
+
+# mkdocs documentation
+/site
+
+# mypy
+.mypy_cache/
+.dmypy.json
+dmypy.json
+
+# Pyre type checker
+.pyre/
+
+# macOS .DS_Store files
+.DS_Store
+
+.python_packages

README.md

@@ -0,0 +1 @@
+# Codedog Demo

app.py

@@ -0,0 +1,26 @@
+import gradio as gr
+
+from codedog_demo.callbacks import get_sample_choices, request_pr_review, show_sample
+
+sample_choices = get_sample_choices()
+
+
+with gr.Blocks(theme="xiaobaiyuan/theme_brief") as demo:
+    gr.Markdown("# Codedog Demo")
+
+    with gr.Tab(label="Try Yourself"):
+        custom_pr_url = gr.Textbox(max_lines=1, placeholder="Paste Github PR URL here", show_label=False)
+        custom_submit = gr.Button(value="Review It")
+        custom_content = gr.Markdown(value="")
+
+        custom_submit.click(request_pr_review, inputs=[custom_pr_url], outputs=[custom_content])
+
+    with gr.Tab(label="Samples"):
+        sample_choice = gr.Radio(choices=sample_choices, type="index", show_label=False)
+        sample_content = gr.Markdown(value="")
+
+        sample_choice.input(show_sample, inputs=[sample_choice], outputs=[sample_content])
+
+
+if __name__ == "__main__":
+    demo.launch()

codedog_demo/callbacks.py

@@ -0,0 +1,38 @@
+from os import environ as env
+from os import listdir
+from typing import List
+
+import requests
+
+from codedog_demo.github_utils import parse_github_pr_url
+
+codedog_api = env.get("CODEDOG_API_URL", "")
+installation = int(env.get("INSTALLATION_ID", 0))
+
+
+sample_names = []
+sample_contents = []
+for file in listdir("samples"):
+    sample_names.append(file.replace("@", "/"))
+    with open("samples/" + file, "r") as f:
+        sample_contents.append(f.read())
+
+
+def request_pr_review(url: str):
+    repo, pr_number = parse_github_pr_url(url)
+    if not repo or not pr_number:
+        return "Invalid URL. Accept format is: https://www.github.com/{owner}/{repository}/pull/{pr_number}"
+
+    response = requests.post(
+        codedog_api, json={"repository": repo, "pull_request_number": pr_number, "installation_id": installation}
+    )
+    result = response.text
+    return result
+
+
+def get_sample_choices() -> List[str]:
+    return sample_names
+
+
+def show_sample(idx: int) -> str:
+    return sample_contents[idx]

codedog_demo/github_utils.py

@@ -0,0 +1,17 @@
+from typing import Tuple
+
+
+def parse_github_pr_url(url: str) -> Tuple[str, int]:
+    # Input = https://github.com/codedog-ai/codedog/pull/11
+    # Output = codedog-ai/codedog, 11
+    if not url:
+        return ("", 0)
+
+    try:
+        parts = url.split("/")
+        owner_repo = parts[3] + "/" + parts[4]
+        pr_number = int(parts[-1])
+    except Exception:
+        return ("", 0)
+
+    return (owner_repo, pr_number)

config.toml

@@ -0,0 +1,2 @@
+[server]
+runOnSave = true

poetry.toml

@@ -0,0 +1,2 @@
+[virtualenvs]
+in-project = true

pyproject.toml

@@ -0,0 +1,17 @@
+[tool.poetry]
+name = "codedog-demo"
+version = "0.1.0"
+description = "Demo pull request review ability"
+authors = ["arcadia822 <arcadia822@gmail.com>"]
+license = "MIT"
+readme = "README.md"
+packages = [{ include = "codedog_demo" }]
+
+[tool.poetry.dependencies]
+python = "^3.10"
+gradio = "^3.39.0"
+
+
+[build-system]
+requires = ["poetry-core"]
+build-backend = "poetry.core.masonry.api"

samples/codedog-ai@codedog #2

@@ -0,0 +1 @@
+# Sample