updated doc-mcp app

app.py

@@ -791,74 +791,125 @@ with gr.Blocks(title="Doc-MCP") as demo:
 
             with gr.Row():
                 with gr.Column(scale=2):
-                    # Repository selection
-                    repo_dropdown = gr.Dropdown(
-                        choices=get_available_repositories(),
-                        label="Select Documentation Repository",
-                        value=None,
-                        interactive=True,
-                        allow_custom_value=False,
-                    )
+                    # Repository selection - Dropdown that becomes textbox when selected
+                    with gr.Row():
+                        repo_dropdown = gr.Dropdown(
+                            choices=get_available_repositories() or ["No repositories available"],
+                            label="📚 Select Documentation Repository",
+                            value=None,
+                            interactive=True,
+                            allow_custom_value=True,
+                            info="Choose from available repositories"
+                        )
+                        
+                        # Hidden textbox that will become visible when repo is selected
+                        selected_repo_textbox = gr.Textbox(
+                            label="🎯 Selected Repository",
+                            value="",
+                            interactive=False,
+                            visible=False,
+                            info="Currently selected repository for querying"
+                        )
+                        
                     refresh_repos_btn = gr.Button(
-                        "🔄 Refresh Repositories", variant="secondary", size="sm"
+                        "🔄 Refresh Repository List", variant="secondary", size="sm"
                     )
 
                     # Query mode selection
                     query_mode = gr.Radio(
                         choices=["default", "text_search", "hybrid"],
-                        label="Query Mode",
+                        label="🔍 Search Strategy",
                         value="default",
-                        info="default: semantic similarity, text_search: keyword-based, hybrid: combines both",
+                        info="• default: Semantic similarity (AI understanding)\n• text_search: Keyword matching\n• hybrid: Combined approach for best results",
                     )
 
                     # Query input
                     query_input = gr.Textbox(
-                        label="Your Query",
-                        placeholder="Ask about the documentation...",
+                        label="💭 Ask About Your Documentation",
+                        placeholder="How do I implement a custom component? What are the available API endpoints? How to configure the system?",
                         lines=3,
+                        info="Ask natural language questions about your documentation"
                     )
 
-                    query_btn = gr.Button("🔍 Search", variant="primary", size="lg")
+                    query_btn = gr.Button("🚀 Search Documentation", variant="primary", size="lg")
 
                     # Response display as text area
                     response_output = gr.Textbox(
-                        label="Response",
-                        value="Your query response will appear here...",
+                        label="🤖 AI Assistant Response",
+                        value="Your AI-powered documentation response will appear here with contextual information and source citations...",
                         lines=10,
                         interactive=False,
+                        info="Generated using Nebius LLM with retrieved documentation context"
                     )
 
                 with gr.Column(scale=2):
-                    gr.Markdown("### Source Nodes (JSON)")
+                    gr.Markdown("### 📖 Source References")
+                    gr.Markdown("View the exact documentation sources used to generate the response, with relevance scores and GitHub links.")
 
                     # Source nodes display as JSON
                     sources_output = gr.JSON(
-                        label="Source Nodes",
+                        label="📎 Source Citations & Metadata",
                         value={
-                            "message": "Source nodes will appear here after querying..."
+                            "message": "Source documentation excerpts with relevance scores will appear here after your query...",
+                            "info": "Each source includes file path, relevance score, and content snippet"
                         },
                     )
 
+            # Event handlers
+            def handle_repo_selection(selected_repo):
+                """Handle repository selection from dropdown"""
+                if not selected_repo or selected_repo in ["No repositories available", ""]:
+                    return (
+                        gr.Dropdown(visible=True),  # Keep dropdown visible
+                        gr.Textbox(visible=False, value=""),  # Hide textbox
+                        gr.Button(interactive=False)  # Disable query button
+                    )
+                else:
+                    return (
+                        gr.Dropdown(visible=False),  # Hide dropdown
+                        gr.Textbox(visible=True, value=selected_repo),  # Show textbox with selected repo
+                        gr.Button(interactive=True)  # Enable query button
+                    )
+
+            def reset_repo_selection():
+                """Reset to show dropdown again"""
+                try:
+                    repos = get_available_repositories() or ["No repositories available"]
+                    return (
+                        gr.Dropdown(choices=repos, value=None, visible=True),  # Show dropdown with refreshed choices
+                        gr.Textbox(visible=False, value=""),  # Hide textbox
+                        gr.Button(interactive=False)  # Disable query button
+                    )
+                except Exception as e:
+                    print(f"Error refreshing repository list: {e}")
+                    return (
+                        gr.Dropdown(choices=["Error loading repositories"], value=None, visible=True),
+                        gr.Textbox(visible=False, value=""),
+                        gr.Button(interactive=False)
+                    )
+
             def get_available_docs_repo():
                 """
-                List the available docs of repositories
+                List the available docs of repositories - should be called first to list out all the available repo docs to chat with
 
                 Returns:
-                    List of repo names
+                    Updated dropdown with available repositories
                 """
                 try:
                     repos = get_available_repositories()
-                    return gr.Dropdown(choices=repos, value=repos[0] if repos else None)
+                    if not repos:
+                        repos = ["No repositories available - Please ingest documentation first"]
+                    return gr.Dropdown(choices=repos, value=None)
                 except Exception as e:
                     print(f"Error refreshing repository list: {e}")
-                    return gr.Dropdown(choices=[], value=None)
+                    return gr.Dropdown(choices=["Error loading repositories"], value=None)
 
             # Simple query handler
             def handle_query(repo: str, mode: str, query: str):
                 """
                 Handle query request - returns raw data from retriever
                 Args:
-                    repo: Selected repository
+                    repo: Selected repository from textbox
                     mode: Query mode (default, text_search, hybrid)
                     query: User's query
                 Returns:
@@ -867,8 +918,8 @@ with gr.Blocks(title="Doc-MCP") as demo:
                 if not query.strip():
                     return {"error": "Please enter a query."}
 
-                if not repo:
-                    return {"error": "Please select a repository."}
+                if not repo or repo in ["No repositories available", "Error loading repositories", ""]:
+                    return {"error": "Please select a valid repository."}
 
                 try:
                     # Import QueryRetriever here to avoid circular imports
@@ -895,8 +946,8 @@ with gr.Blocks(title="Doc-MCP") as demo:
                 This function is designed to support Retrieval-Augmented Generation (RAG) by extracting
                 the most relevant context chunks from indexed documentation sources.
                 Args:
-                    repo: Selected repository
-                    mode: Query mode
+                    repo: Selected repository from the textbox input
+                    mode: Query mode (default, text_search, hybrid)
                     query: User's query
                 Returns:
                     Tuple of (response_text, source_nodes_json)
@@ -914,24 +965,42 @@ with gr.Blocks(title="Doc-MCP") as demo:
 
                 return response_text, source_nodes
 
+            # Wire up events
+            
+            # Handle repository selection from dropdown
+            repo_dropdown.change(
+                fn=handle_repo_selection,
+                inputs=[repo_dropdown],
+                outputs=[repo_dropdown, selected_repo_textbox, query_btn],
+                show_api=False
+            )
+            
+            # Handle refresh button - resets to dropdown view
+            refresh_repos_btn.click(
+                fn=reset_repo_selection,
+                outputs=[repo_dropdown, selected_repo_textbox, query_btn],
+                show_api=False
+            )
+            
+            # Also provide API endpoint for listing repositories
             refresh_repos_btn.click(
                 fn=get_available_docs_repo,
                 outputs=[repo_dropdown],
-                api_name="List available docs",
+                api_name="list_available_docs",
             )
 
-            # Simple event wiring - single button click
+            # Query button uses the textbox value (not dropdown)
             query_btn.click(
                 fn=make_query,
-                inputs=[repo_dropdown, query_mode, query_input],
+                inputs=[selected_repo_textbox, query_mode, query_input],  # Use textbox, not dropdown
                 outputs=[response_output, sources_output],
-                api_name="Query docs",
+                api_name="query_documentation",
             )
 
             # Also allow Enter key to trigger query
             query_input.submit(
                 fn=make_query,
-                inputs=[repo_dropdown, query_mode, query_input],
+                inputs=[selected_repo_textbox, query_mode, query_input],  # Use textbox, not dropdown
                 outputs=[response_output, sources_output],
                 show_api=False,
             )
@@ -1149,4 +1218,4 @@ with gr.Blocks(title="Doc-MCP") as demo:
 
 
 if __name__ == "__main__":
-    demo.launch(mcp_server=True, share=True)
+    demo.launch(mcp_server=True)

rag/query.py

@@ -60,8 +60,6 @@ class QueryRetriever:
             response_mode="refine",
         )
 
-        
-
         response = query_engine.query(query)
         nodes = []
         for node in response.source_nodes: