Update repo_explorer.py

repo_explorer.py

@@ -154,6 +154,7 @@ def create_repo_explorer_tab() -> Tuple[Dict[str, gr.components.Component], Dict
             )
         with gr.Column(scale=1):
             load_repo_btn = gr.Button("🚀 Load Repository", variant="primary", size="lg")
+            visit_hf_btn = gr.Button("🔗 Visit on HF", variant="secondary", size="lg", visible=False)
     
     with gr.Row():
         repo_status_display = gr.Textbox(
@@ -201,6 +202,7 @@ def create_repo_explorer_tab() -> Tuple[Dict[str, gr.components.Component], Dict
     components = {
         "repo_explorer_input": repo_explorer_input,
         "load_repo_btn": load_repo_btn,
+        "visit_hf_btn": visit_hf_btn,
         "repo_status_display": repo_status_display,
         "repo_chatbot": repo_chatbot,
         "repo_msg_input": repo_msg_input,
@@ -290,10 +292,16 @@ Answer the user's question based on your comprehensive knowledge of this reposit
     
     return history
 
-def handle_load_repository_with_vectorization(repo_id: str) -> Tuple[str, str]:
+def get_huggingface_url(repo_id: str) -> str:
+    """Generate the Hugging Face URL for a repository."""
+    if not repo_id.strip():
+        return ""
+    return f"https://huggingface.co/{repo_id}"
+
+def handle_load_repository_with_vectorization(repo_id: str) -> Tuple[str, str, gr.Button]:
     """Load repository and create both context summary and vector embeddings."""
     if not repo_id.strip():
-        return "Status: Please enter a repository ID.", ""
+        return "Status: Please enter a repository ID.", "", gr.Button(visible=False)
     
     try:
         logger.info(f"Loading repository with vectorization: {repo_id}")
@@ -305,7 +313,7 @@ def handle_load_repository_with_vectorization(repo_id: str) -> Tuple[str, str]:
         except Exception as e:
             logger.error(f"Error downloading repository {repo_id}: {e}")
             error_status = f"❌ Error downloading repository: {e}"
-            return error_status, ""
+            return error_status, "", gr.Button(visible=False)
         
         # Read the combined content
         with open(combined_text_path, "r", encoding="utf-8") as f:
@@ -324,13 +332,16 @@ def handle_load_repository_with_vectorization(repo_id: str) -> Tuple[str, str]:
         else:
             status = f"✅ Repository '{repo_id}' loaded successfully!\n📁 Files processed and ready for exploration.\n⚠️ Vectorization failed - using text-only analysis.\n💬 You can now ask questions about this repository."
         
+        # Show the HF visit button since repository loaded successfully
+        updated_button = gr.Button("🔗 Visit on HF", variant="secondary", size="lg", visible=True)
+        
         logger.info(f"Repository {repo_id} loaded and processed successfully")
-        return status, context_summary
+        return status, context_summary, updated_button
         
     except Exception as e:
         logger.error(f"Error loading repository {repo_id}: {e}")
         error_status = f"❌ Error loading repository: {e}"
-        return error_status, ""
+        return error_status, "", gr.Button(visible=False)
 
 def initialize_repo_chatbot(repo_status: str, repo_id: str, repo_context_summary: str) -> List[Dict[str, str]]:
     """Initialize the repository chatbot with a welcome message after successful repo loading."""
@@ -352,7 +363,7 @@ def setup_repo_explorer_events(components: Dict[str, gr.components.Component], s
     components["load_repo_btn"].click(
         fn=handle_load_repository_with_vectorization,
         inputs=[components["repo_explorer_input"]],
-        outputs=[components["repo_status_display"], states["repo_context_summary"]]
+        outputs=[components["repo_status_display"], states["repo_context_summary"], components["visit_hf_btn"]]
     ).then(
         fn=lambda repo_id: repo_id,
         inputs=[components["repo_explorer_input"]],
@@ -363,6 +374,14 @@ def setup_repo_explorer_events(components: Dict[str, gr.components.Component], s
         outputs=[components["repo_chatbot"]]
     )
     
+    # Visit Hugging Face repository button - opens in new tab
+    components["visit_hf_btn"].click(
+        fn=None,
+        inputs=[states["current_repo_id"]],
+        outputs=[],
+        js="(repo_id) => { if(repo_id) window.open('https://huggingface.co/' + repo_id, '_blank'); }"
+    )
+    
     # Chat message submission events
     components["repo_msg_input"].submit(
         fn=handle_repo_user_message,