Update app.py

app.py

@@ -40,8 +40,9 @@ def initialize_environment():
 
 # GitHub API handler
 class GitHubAPI:
-    def __init__(self, token: str):
+    def __init__(self, token: str, logger: logging.Logger):
         self.token = token
+        self.logger = logger
         self.headers = {
             'Authorization': f'token {token}',
             'Accept': 'application/vnd.github.v3+json'
@@ -59,12 +60,12 @@ class GitHubAPI:
             if remaining < 10:
                 wait_time = max(0, reset_time - int(time.time()))
                 if wait_time > 0:
-                    logger.warning(f"Rate limit nearly exceeded. Waiting {wait_time} seconds before retrying...")
+                    self.logger.warning(f"Rate limit nearly exceeded. Waiting {wait_time} seconds before retrying...")
                     time.sleep(wait_time)
                 return False
             return True
         except requests.exceptions.RequestException as e:
-            logger.error(f"Error checking rate limit: {str(e)}. Retrying...")
+            self.logger.error(f"Error checking rate limit: {str(e)}. Retrying...")
             return True
 
     def get_repository(self, owner: str, repo: str) -> Dict:
@@ -73,10 +74,10 @@ class GitHubAPI:
             response.raise_for_status()
             return response.json()
         except requests.HTTPError as e:
-            logger.error(f"HTTP error getting repository info for {owner}/{repo}: {str(e)}. Please check the repository details.")
+            self.logger.error(f"HTTP error getting repository info for {owner}/{repo}: {str(e)}. Please check the repository details.")
             raise
         except Exception as e:
-            logger.error(f"Error getting repository info: {str(e)}")
+            self.logger.error(f"Error getting repository info: {str(e)}")
             raise
 
     def get_issues(self, owner: str, repo: str, state: str = 'open') -> List[Dict]:
@@ -89,23 +90,24 @@ class GitHubAPI:
             issues = response.json()
             return [issue for issue in issues if 'pull_request' not in issue]
         except Exception as e:
-            logger.error(f"Error fetching issues for repository {owner}/{repo}: {str(e)}. Please verify the repository and token.")
+            self.logger.error(f"Error fetching issues for repository {owner}/{repo}: {str(e)}. Please verify the repository and token.")
             return []
 
 # GitHub Bot
 class GitHubBot:
-    def __init__(self):
+    def __init__(self, logger: logging.Logger):
         self.github_api = None
+        self.logger = logger
 
     def initialize_api(self, token: str):
-        self.github_api = GitHubAPI(token)
+        self.github_api = GitHubAPI(token, self.logger)
 
     def fetch_issues(self, token: str, owner: str, repo: str) -> List[Dict]:
         try:
             self.initialize_api(token)
             return self.github_api.get_issues(owner, repo)
         except Exception as e:
-            logger.error(f"Error fetching issues for repository {owner}/{repo}: {str(e)}")
+            self.logger.error(f"Error fetching issues for repository {owner}/{repo}: {str(e )}")
             return []
 
     def resolve_issue(self, token: str, owner: str, repo: str, issue_number: int, resolution: str, forked_repo: str) -> str:
@@ -120,6 +122,10 @@ class GitHubBot:
             with open(resolution_file, "w") as f:
                 f.write(f"# Resolution for Issue #{issue_number}\n\n{resolution}")
 
+            # Validate forked_repo before cloning
+            if not forked_repo:
+                raise ValueError("Forked repository URL cannot be empty.")
+
             # Clone the forked repo
             subprocess.run(['git', 'clone', forked_repo, '/tmp/' + forked_repo.split('/')[-1]], check=True)
 
@@ -141,7 +147,7 @@ class GitHubBot:
 
         except Exception as e:
             error_msg = f"Error resolving issue #{issue_number} in repository {owner}/{repo}: {str(e)}"
-            logger.error(error_msg)
+            self.logger.error(error_msg)
             return error_msg
 
     def suggest_automated_fixes(self, issue_title: str) -> str:
@@ -150,7 +156,7 @@ class GitHubBot:
         return "No automated fix available for this issue."
 
 def handle_issue_selection(token, owner, repo, issue_number, resolution, forked_repo):
-    bot = GitHubBot()
+    bot = GitHubBot(logger)
     result = bot.resolve_issue(token, owner, repo, issue_number, resolution, forked_repo)
     return result
 
@@ -182,7 +188,7 @@ def extract_info_from_url(url: str) -> Dict[str, Any]:
     return info
 
 # Initialize GitHubBot globally
-bot = GitHubBot()
+bot = GitHubBot(logger)
 
 # Define missing functions with validation
 def fetch_issues(token, repo_url):
@@ -261,6 +267,7 @@ def cleanup():
 def main():
     # Initialize environment and logger
     initialize_environment()
+    global logger
     logger = initialize_logger()
 
     # Register cleanup handlers