app.py

import datetime
import os
import shutil

import gradio as gr
import requests

from utils import build_issue_dict
from utils import build_embeddings
from utils.defaults import OWNER, REPO
from utils.fetch import get_issues
from utils.find_similar_issues import get_similar_issues
from utils.update_stored_issues import update_issues


def get_query_issue_information(issue_no, token):
    headers = {
        "Accept": "application/vnd.github+json",
        "Authorization": f"{token}",
        "X-GitHub-Api-Version": "2022-11-28",
        "User-Agent": "amyeroberts",
    }
    request = requests.get(
        f"https://api.github.com/repos/{OWNER}/{REPO}/issues/{issue_no}",
        headers=headers,
    )
    if request.status_code != 200:
        raise ValueError(f"Request failed with status code {request.status_code} and message {request.text}")

    return request.json()


def run_find_similar_issues(token, n_issues, issue_no, query, issue_types):
    if issue_no == "":
        issue_no = None
    if query == "":
        query = None

    if len(issue_types) == 0:
        raise ValueError("At least one issue type must be selected")

    similar_issues = []
    similar_pulls = []
    if "Issue" in issue_types:
        similar_issues = get_similar_issues(issue_no=issue_no, query=query, token=token, top_k=n_issues, issue_type="issue")
    if "Pull Request" in issue_types:
        similar_pulls = get_similar_issues(issue_no=issue_no, query=query, token=token, top_k=n_issues, issue_type="pull")

    issues_html = [f"<a href='{issue['html_url']}' target='_blank'>#{issue['number']} - {issue['title']}</a>" for issue in similar_issues]
    issues_html = "<br>".join(issues_html)

    pulls_html = [f"<a href='{issue['html_url']}' target='_blank'>#{issue['number']} - {issue['title']}</a>" for issue in similar_pulls]
    pulls_html = "<br>".join(pulls_html)

    final = ""
    if len(issues_html) > 0:
        final += f"<h2>Issues</h2>{issues_html}"
    if len(pulls_html) > 0:
        final += f"<h2>Pull Requests</h2>{pulls_html}"

    # return issues_html
    return final


def update(token):
    # Archive the stored issues
    if os.path.exists("issues.json"):
        date_time = datetime.datetime.now().strftime("%Y-%m-%d_%H-%M-%S")
        shutil.copy("issues.json", f"{date_time}_issues.json")

    # Retrieve new issues
    get_issues(overwrite=False, update=True, output_filename="issues.json", token=token)
    # Update any issues that have been updated since the last update
    update_issues(token=token)
    # Update the dictionary of issues
    build_issue_dict.build_json_file("issues.json", "issues_dict.json")
    # Update the embeddings
    build_embeddings.embed_issues(
        input_filename="issues_dict.json",
        issue_type="issue",
        model_id="all-mpnet-base-v2",
        update=True
    )
    build_embeddings.embed_issues(
        input_filename="issues_dict.json",
        issue_type="pull",
        model_id="all-mpnet-base-v2",
        update=True
    )


with gr.Blocks(title="Github Bot") as demo:
    with gr.Tab("Find similar issues"):
        with gr.Row():
            gr.Markdown("## Query settings")
        with gr.Row():
            gr.Markdown("Configure the settings for the query. You can either search for similar issues to a given issue or search for issues based on a query.")
        with gr.Row():
            with gr.Column():
                gr.Markdown("Find similar issues to a given issue or query")
                issue_no = gr.Textbox(label="Github Issue", placeholder="Github issue you want to find similar issues to")
                query = gr.Textbox(label="Query", placeholder="Search for issues")
            with gr.Column():
                token = gr.Textbox(label="Github Token", placeholder="Your github token for authentication. This is not stored anywhere.")
                n_issues = gr.Slider(1, 50, value=5, step=1, label="Number of similar issues", info="Choose between 1 and 50")
                issue_types = gr.CheckboxGroup(["Issue", "Pull Request"], label="Issue types")

        with gr.Row():
                submit_button = gr.Button(value="Submit")

        with gr.Row():
            with gr.Row():
                issues_html = gr.HTML(label="Issue text", elem_id="issue_html")
                submit_button.click(run_find_similar_issues, outputs=[issues_html], inputs=[token, n_issues, issue_no, query, issue_types])

        with gr.Row():
            with gr.Column():
                gr.Markdown("Press `Update Button` to update the stored issues. This will take a few minutes.")
                update_button = gr.Button(value="Update issues")
                update_button.click(update, outputs=[], inputs=[token], trigger_mode="once")
            with gr.Column():
                pass

    with gr.Tab("Find maintainers to ping"):
        gr.Markdown("Find maintainers to ping for a given issue or pull request - DO NOT CURRENTLY HAVE FUNCTIONALITY")
        with gr.Row():
            issue = gr.Textbox(label="Github Issue / PR", placeholder="Issue or PR you want to find maintainers to ping for")
        with gr.Row():
            token = gr.Textbox(label="Github Token", placeholder="Your github token for authentication. This is not stored anywhere.")


if __name__ == "__main__":
    demo.launch()