Update app.py

app.py

@@ -14,7 +14,6 @@ import webbrowser
 import urllib.parse
 import warnings
 import torch
-import spaces
 from accelerate import Accelerator
 
 warnings.filterwarnings('ignore', category=UserWarning, module='__main__')
@@ -123,20 +122,20 @@ class GitHubBot:
         """Initialize GitHub API with token"""
         self.github_api = GitHubAPI(token)
 
-    def fetch_issues(self, token: str, username: str, repo: str) -> List[Dict]:
+    def fetch_issues(self, token: str, owner: str, repo: str) -> List[Dict]:
         """Fetch issues from GitHub repository"""
         try:
             self.initialize_api(token)
-            return self.github_api.get_issues(username, repo)
+            return self.github_api.get_issues(owner, repo)
         except Exception as e:
             logger.error(f"Error fetching issues: {str(e)}")
             return []
 
-    def resolve_issue(self, token: str, username: str, repo: str, issue_number: int, resolution: str, forked_repo: str) -> str:
+    def resolve_issue(self, token: str, owner: str, repo: str, issue_number: int, resolution: str, forked_repo: str) -> str:
         """Resolve a GitHub issue and submit PR."""
         try:
             self.initialize_api(token)
-            self.github_api.get_repository(username, repo)
+            self.github_api.get_repository(owner, repo)
 
             # Create resolution file
             timestamp = datetime.now().strftime("%Y%m%d_%H%M%S")
@@ -160,7 +159,7 @@ class GitHubBot:
             subprocess.run(['git', 'push', 'origin', 'HEAD'], check=True)
 
             # Open Pull Request page
-            webbrowser.open(f'https://github.com/{forked_repo.split("/")[-1]}/compare/master...{username}:{forked_repo.split("/")[-1]}_resolved_issue_{issue_number}')
+            webbrowser.open(f'https://github.com/{forked_repo.split("/")[-1]}/compare/master...{owner}:{forked_repo.split("/")[-1]}_resolved_issue_{issue_number}')
 
             return f"Resolution saved: {resolution_file}"
 
@@ -169,9 +168,9 @@ class GitHubBot:
             logger.error(error_msg)
             return error_msg
 
-def handle_issue_selection(token, username, repo, issue_number, resolution, forked_repo):
+def handle_issue_selection(token, owner, repo, issue_number, resolution, forked_repo):
     bot = GitHubBot()
-    result = bot.resolve_issue(token, username, repo, issue_number, resolution, forked_repo)
+    result = bot.resolve_issue(token, owner, repo, issue_number, resolution, forked_repo)
     return result
 
 custom_css = """
@@ -255,14 +254,9 @@ def create_gradio_interface(input_file, output_directory):
                 type="password",
                 elem_classes="input input-bordered input-primary"
             )
-            github_username = gr.Textbox(
-                label="Repository Owner",
-                placeholder="Enter repository owner username",
-                elem_classes="input input-bordered input-primary"
-            )
-            github_repo = gr.Textbox(
-                label="Repository Name",
-                placeholder="Enter repository name",
+            repo_link = gr.Textbox(
+                label="Repository Link",
+                placeholder="Enter repository link (e.g., https://github.com/owner/repo)",
                 elem_classes="input input-bordered input-primary"
             )
 
@@ -303,11 +297,12 @@ def create_gradio_interface(input_file, output_directory):
 
         gr.HTML("</div></div>")  # Close card and container divs
 
-        def fetch_issues_handler(token, username, repo):
-            if not all([token, username, repo]):
+        def fetch_issues_handler(token, repo_link):
+            if not all([token, repo_link]):
                 return gr.Dropdown(choices=["Please provide all required fields"])
 
-            issues = bot.fetch_issues(token, username, repo)
+            owner, repo = repo_link.split('/')[-2:]
+            issues = bot.fetch_issues(token, owner, repo)
             if not issues:
                 return gr.Dropdown(choices=["No issues found or error occurred"])
 
@@ -315,20 +310,21 @@ def create_gradio_interface(input_file, output_directory):
                 choices=[f"#{issue['number']}: {issue['title']}" for issue in issues]
             )
 
-        def resolve_issue_handler(token, username, repo, selected_issue, resolution, forked_repo):
+        def resolve_issue_handler(token, repo_link, selected_issue, resolution, forked_repo):
             if not selected_issue:
                 return "Please select an issue first"
 
             try:
+                owner, repo = repo_link.split('/')[-2:]
                 issue_number = int(selected_issue.split(':')[0][1:])
-                return handle_issue_selection(token, username, repo, issue_number, resolution, forked_repo)
+                return handle_issue_selection(token, owner, repo, issue_number, resolution, forked_repo)
             except Exception as e:
                 return f"Error: {str(e)}"
 
         # Connect components
         fetch_button.click(
             fetch_issues_handler,
-            inputs=[github_token, github_username, github_repo],
+            inputs=[github_token, repo_link],
             outputs=[issue_dropdown]
         )
 
@@ -336,8 +332,7 @@ def create_gradio_interface(input_file, output_directory):
             resolve_issue_handler,
             inputs=[
                 github_token,
-                github_username,
-                github_repo,
+                repo_link,
                 issue_dropdown,
                 resolution_text,
                 forked_repo