update demo

app.py

@@ -9,11 +9,25 @@ 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.Row():
+            custom_pr_url = gr.Textbox(
+                max_lines=1,
+                value="https://github.com/codedog-ai/codedog/pull/2",
+                placeholder="Paste Github PR URL here",
+                show_label=False,
+            )
+        with gr.Row():
+            custom_submit = gr.Button(value="Review It")
+        with gr.Row():
+            with gr.Tab(label="Markdown"):
+                custom_content = gr.Markdown(value="")
+            with gr.Tab(label="Raw"):
+                custom_content_raw = gr.Textbox(show_label=False)
+        custom_submit.click(
+            request_pr_review,
+            inputs=[custom_pr_url],
+            outputs=[custom_content, custom_content_raw],
+        )
 
     with gr.Tab(label="Samples"):
         sample_choice = gr.Radio(choices=sample_choices, type="index", show_label=False)

codedog_demo/callbacks.py

@@ -1,14 +1,17 @@
+import time
+import traceback
+from functools import lru_cache
 from os import environ as env
 from os import listdir
 from typing import List
 
+import gradio as gr
 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))
-
+github_token = env.get("GITHUB_TOKEN", "")
 
 sample_names = []
 sample_contents = []
@@ -19,17 +22,38 @@ for file in listdir("samples"):
 
 
 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}"
+    try:
+        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}", ""
+
+        result = _request_pr_review(repo, pr_number, ttl_hash=get_ttl_hash())
+
+    except Exception:
+        traceback.print_exc()
+        return "Something went wrong. Please try again later.", ""
+    return result, result
 
+
+@lru_cache(maxsize=100)
+def _request_pr_review(repo: str, pr_number: int, ttl_hash=None):
     response = requests.post(
-        codedog_api, json={"repository": repo, "pull_request_number": pr_number, "installation_id": installation}
+        codedog_api, json={"repository": repo, "pull_request_number": pr_number, "token": github_token}
     )
     result = response.text
+    if len(result) < 100:
+        if result == "stream timeout":
+            raise ValueError("Timeout")
+        print(f"Error result: {result}")
+        raise ValueError()
     return result
 
 
+def get_ttl_hash(seconds=120):
+    """Return the same value withing `seconds` time period"""
+    return round(time.time() / seconds)
+
+
 def get_sample_choices() -> List[str]:
     return sample_names