feat: support branches

app.py

@@ -63,9 +63,9 @@ def authenticate(password):
 def analyze_repos(repo_urls, role, seniority, assignment_details, focus_areas, password):
     auth_error, auth_success = authenticate(password)
     if auth_error:
-        return auth_error, gr.update(visible=False)  # If incorrect password, return error
-        
-    """Clone and analyze multiple GitHub repositories."""
+        return auth_error, gr.update(visible=False)
+
+    """Clone and analyze multiple GitHub repositories, supporting specific branches."""
     repo_urls = [url.strip() for url in repo_urls.split(",") if url.strip()]
     
     if not all(url.startswith("https://github.com/") for url in repo_urls):
@@ -79,16 +79,24 @@ def analyze_repos(repo_urls, role, seniority, assignment_details, focus_areas, p
         progress = gr.update(value="🔄 Cloning repositories...", visible=True)
         
         for repo_url in repo_urls:
+            # Extract branch if provided (URLs with "/tree/")
+            branch = None
+            if "/tree/" in repo_url:
+                repo_url, branch = repo_url.split("/tree/")[:2]
+                repo_url = repo_url.rstrip("/")  # Clean base repo URL
+            
             repo_name = repo_url.split("/")[-1]
             repo_path = os.path.join(temp_dir, repo_name)
-            
+
             try:
-                git.Repo.clone_from(repo_url, repo_path)
+                repo = git.Repo.clone_from(repo_url + ".git", repo_path)
+                if branch:
+                    repo.git.checkout(branch)
             except Exception as e:
-                repos_data += f"\n❌ Failed to clone `{repo_name}`: {str(e)}\n"
+                repos_data += f"\n❌ Failed to clone `{repo_name}` (Branch: `{branch or 'default'}`): {str(e)}\n"
                 continue
 
-            repo_data = f"\n📂 **Repository: {repo_name}**\n"
+            repo_data = f"\n📂 **Repository: {repo_name}** (Branch: `{branch or 'default'}`)\n"
             file_count = 0  
 
             for root, _, filenames in os.walk(repo_path):
@@ -134,7 +142,7 @@ def analyze_repos(repo_urls, role, seniority, assignment_details, focus_areas, p
     
     finally:
         shutil.rmtree(temp_dir)  # Cleanup
-
+        
 # Gradio UI
 with gr.Blocks() as app:
     gr.Markdown("# 🛠️ AI-Powered Candidate Evaluation System")