Refactor stuff

app.py

@@ -84,6 +84,12 @@ tickets = []
 
 repo_name = gr.State()
 git_tickets = gr.State()
+git_titles = gr.State()
+git_ticket_choices = gr.State()
+git_tickets.value = []
+git_titles.value = []
+git_ticket_choices.value = []
+
 # repoName = "https://github.com/gradio-app/gradio.git"
 
 embeddings = OpenAIEmbeddings(disallowed_special=())
@@ -190,7 +196,6 @@ def index_repo(textbox: str, dropdown: str) -> Response:
             stdout="",
         )
 
-    # global ticket_choices, ticket_titles, tickets
     repo = "/".join(repo[:-4].split("/")[-2:])
     tickets = fetchGithubIssues(repo, 10)
     git_tickets.value = tickets
@@ -206,7 +211,6 @@ def index_repo(textbox: str, dropdown: str) -> Response:
 
 def answer_questions(question: str, github: str, **kwargs) -> Response:
 
-    # global repoName
     repoName = repo_name.value
     github = repoName[:-4]
     print("REPO NAME: " + github)
@@ -258,11 +262,11 @@ def answer_questions(question: str, github: str, **kwargs) -> Response:
     )
 
 
-def fetchGithubIssues(repo: str, num_issues:int, **kwargs) -> Response:
+def fetchGithubIssues(num_issues:int, **kwargs) -> Response:
     """
         This endpoint should get a list of all the github issues that are open for this repository
     """
-
+    repo = repo_name.value
     batch = []
     all_issues = []
     per_page = 100  # Number of issues to return per page
@@ -288,6 +292,10 @@ def fetchGithubIssues(repo: str, num_issues:int, **kwargs) -> Response:
                 "comments_url": issue["comments_url"],
             })
 
+    # This should set the state variables for tickets
+    git_tickets.value = issues_data
+    git_ticket_choices.value = {git_tickets.value["title"]: ticket for ticket in tickets}
+    git_titles.value = [git_tickets.value["title"] for ticket in tickets]
     return  issues_data
 
 
@@ -368,13 +376,13 @@ def solveGithubIssue(ticket, history) -> Response:
     """
     # global repoName
     repoName = repo_name.value
-    ticket_choices = git_tickets.value
+    # ticket_choices = git_tickets.value
     # global ticket_choices
     github = repoName[:-4]
 
     repoFolder = github.split("/")[-1]
-    body = ticket_choices[ticket]["body"]
-    title = ticket_choices[ticket]["title"]
+    body = git_ticket_choices.value[ticket]["body"]
+    title = git_ticket_choices.value[ticket]["title"]
     question = """
         Given the code in the {} repo, propose a solution for this ticket {} that includes a
         high level implementation, narrowing down the root cause of the issue and psuedocode if
@@ -487,9 +495,6 @@ with gr.Blocks() as demo:
 
     success_response = gr.Textbox(label="")
     ingest_btn = gr.Button("Index repo")
-    # ticketDropdown = gr.Dropdown()
-
-    # repoTextBox.submit(fetchGithubIssues, [], ticketDropdown)
 
     with gr.Column(visible=False) as launch_product:
 
@@ -509,35 +514,30 @@ with gr.Blocks() as demo:
         with gr.Tab("Bug Triage"):
 
             # Display the titles in the dropdown
-            def create_ticket_dropdown(tickets):
+            def create_ticket_dropdown():
+
+                ticket_choices = {git_tickets.value["title"]: ticket for ticket in tickets}
+                ticket_titles = [git_tickets.value["title"] for ticket in tickets]
 
+                # return ticket_choices, ticket_titles
                 return gr.Dropdown.update(
                     choices=ticket_titles, value=ticket_titles[0]
                 ), gr.update(visible=True)
 
             # global ticket_choices, ticket_titles, tickets
-            repo = "/".join(repo_name.value[:-4].split("/")[-2:])
-            tickets = fetchGithubIssues(repo, 10)
+            # repo = "/".join(repo_name.value[:-4].split("/")[-2:])
+            # tickets = fetchGithubIssues(repo, 10)
 
             # Create the dropdown
-            ticket_choices = {ticket["title"]: ticket for ticket in tickets}
-            ticket_titles = [ticket["title"] for ticket in tickets]
-
-            # Here you want to first call the getGithubIssues function
-            # repo = gr.Interface.get_session_state("repo")
-            #   repo = "/".join(repoName[:-4].split("/")[-2:])
-            #   tickets = fetchGithubIssues(repo, 10)
-
-            #   # Create the dropdown
-            #   global ticket_choices
-            ticket_choices = {ticket["title"]: ticket for ticket in tickets}
-            ticket_titles = [ticket["title"] for ticket in tickets]
-
-            ticketDropdown = gr.Dropdown(choices=ticket_titles, title="Github Issues")
+            # ticket_choices = {git_tickets.value["title"]: ticket for ticket in tickets}
+            # ticket_titles = [git_tickets.value["title"] for ticket in tickets]
+            ingestTickets = gr.Button("Ingest github tickets").submit(fetchGithubIssues, inputs=[20], outputs=[]).then(create_ticket_dropdown)
+            ticketDropdown = gr.Dropdown(choices=git_titles.value, title="Github Issues")
 
             # Extract the ticket title, body for the selected ticket
             chatbot = gr.Chatbot()
             msg = gr.Textbox()
+
             clear = gr.Button("Clear")
 
             ticketDropdown.change(solveGithubIssue, inputs=[ticketDropdown, chatbot], outputs=[chatbot])
@@ -650,7 +650,7 @@ with gr.Blocks() as demo:
             )
 
 
-    ingest_btn.click(fn=index_repo, inputs=[repoTextBox, ingestedRepos], outputs=[success_response, launch_product], api_name="index_repo")
+    ingest_btn.click(fn=index_repo, inputs=[repoTextBox, ingestedRepos], outputs=[success_response, launch_product], api_name="index_repo").then(fn=fetchGithubIssues, inputs=[20], outputs=[])
     # ingest_btn.click(fn=)