Update app.py

app.py

@@ -12,7 +12,7 @@ model = AutoModelForSeq2SeqLM.from_pretrained(model_name)
 
 # Constants for enhanced organization
 GITHUB_API_BASE_URL = "https://api.github.com/repos"
-DEFAULT_MODEL = "microsoft/CodeBERT-base"
+DEFAULT_MODEL = "apple/OpenELM"
 MAX_RELATED_ISSUES = 3
 
 # Load a pre-trained model for sentence similarity
@@ -95,6 +95,7 @@ class MyChatbot(gr.Chatbot):
     def __init__(self, fn, **kwargs):
         super().__init__(fn, **kwargs)
         self.issues = []  # Store fetched issues
+        self.current_issue = None  # Store the currently selected issue
 
     def postprocess(self, y):
         """Post-processes the response to handle commands and display results."""
@@ -123,18 +124,26 @@ class MyChatbot(gr.Chatbot):
         elif y['command'].isdigit() and self.issues:
             try:
                 issue_number = int(y['command']) - 1
-                issue = self.issues[issue_number]
-                issue_text = issue['title'] + "\n\n" + issue['body']
+                self.current_issue = self.issues[issue_number]  # Store the selected issue
+                issue_text = self.current_issue['title'] + "\n\n" + self.current_issue['body']
                 resolution = analyze_issues(issue_text, y['selected_model'], y['severity'], y['programming_language'])
                 related_issues = find_related_issues(issue_text, self.issues)
                 related_issue_text = "\n".join(
                     [f"- {issue['title']} (Similarity: {similarity:.2f})" for issue, similarity in related_issues]
                 )
-                return f"Resolution for Issue '{issue['title']}':\n{resolution['assistant_response']}\n\nRelated Issues:\n{related_issue_text}"
+                return f"Resolution for Issue '{self.current_issue['title']}':\n{resolution['assistant_response']}\n\nRelated Issues:\n{related_issue_text}"
             except Exception as e:
                 return f"Error analyzing issue: {e}"
+        elif y['command'].startswith("/"):
+            # Handle commands like `/generate_code`, `/explain_concept`, etc.
+            if self.current_issue:
+                # Use the current issue's context for these commands
+                issue_text = self.current_issue['title'] + "\n\n" + self.current_issue['body']
+                return analyze_issues(issue_text, y['selected_model'], y['severity'], y['programming_language'])['assistant_response']
+            else:
+                return "Please select an issue first using `/github`."
         else:
-            # For other commands or free-form text, simply display the assistant's response
+            # For free-form text, simply display the assistant's response
             return assistant_response
 
 with gr.Blocks() as demo:
@@ -150,9 +159,9 @@ with gr.Blocks() as demo:
 
     model_dropdown = gr.Dropdown(
         choices=[
-            "microsoft/CodeBERT-base",
-            "Salesforce/codegen-45M-mono",
-            
+            "‎apple/OpenELM",
+            "Gabriel/Swe-review-setfit-model",
+            "acecalisto3/InstructiPhi"
         ],
         label="Select Model for Issue Resolution",
         value=DEFAULT_MODEL,
@@ -166,22 +175,9 @@ with gr.Blocks() as demo:
 
     programming_language_textbox = gr.Textbox(label="Programming Language")
 
-    command_dropdown = gr.Dropdown(
-        choices=[
-            "/github",
-            "/help",
-            "/generate_code",
-            "/explain_concept",
-            "/write_documentation",
-            "/translate_code",
-        ],
-        label="Select Command",
-    )
-
     chatbot = MyChatbot(
         respond,
         additional_inputs=[
-            command_dropdown,
             system_message,
             gr.Slider(minimum=1, maximum=8192, value=2048, step=1, label="Max new tokens"),
             gr.Slider(minimum=0.1, maximum=4.0, value=0.71, step=0.1, label="Temperature"),
@@ -201,6 +197,17 @@ with gr.Blocks() as demo:
         ],
     )
 
+    # Add a button to fetch GitHub issues
+    fetch_issues_button = gr.Button(label="Fetch Issues")
+    fetch_issues_button.click(fn=lambda github_api_token, github_username, github_repository: chatbot.issues, inputs=[github_api_token, github_username, github_repository], outputs=[chatbot])
+
+    # Add a dropdown to select an issue
+    issue_dropdown = gr.Dropdown(label="Select Issue", choices=[], interactive=True)
+    issue_dropdown.change(fn=lambda issue_number, chatbot: chatbot.postprocess(issue_number), inputs=[issue_dropdown, chatbot], outputs=[chatbot])
+
+    # Connect the chatbot input to the issue dropdown
+    chatbot.input.change(fn=lambda chatbot, github_api_token, github_username, github_repository: chatbot.postprocess("/github"), inputs=[chatbot, github_api_token, github_username, github_repository], outputs=[chatbot])
+
 if __name__ == "__main__":
     demo.queue().launch(
         share=True,