Rajesh Betkiker
commited on
Commit
Β·
a7c9a13
1
Parent(s):
0321eee
Added app.py
Browse files
app.py
ADDED
|
@@ -0,0 +1,151 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
"""
|
| 2 |
+
This file is the main file for the hackathon.
|
| 3 |
+
It contains the Gradio interface for the hackathon as a MCP server.
|
| 4 |
+
It exposes the following tools:
|
| 5 |
+
- search_knowledge_base_for_context
|
| 6 |
+
- research_write_review_topic
|
| 7 |
+
|
| 8 |
+
"""
|
| 9 |
+
from dotenv import load_dotenv
|
| 10 |
+
load_dotenv() # Load environment variables from .env file
|
| 11 |
+
|
| 12 |
+
import logging
|
| 13 |
+
|
| 14 |
+
# Configure logging to write to a file instead of stdout/stderr
|
| 15 |
+
# This avoids interference with the MCP communication channel
|
| 16 |
+
logging.basicConfig(
|
| 17 |
+
filename='hackathon-mcp.log', # Log to a file instead of stdout/stderr
|
| 18 |
+
level=logging.INFO,
|
| 19 |
+
format='%(asctime)s - %(name)s - %(levelname)s - %(message)s'
|
| 20 |
+
)
|
| 21 |
+
logger = logging.getLogger(__name__)
|
| 22 |
+
|
| 23 |
+
import gradio as gr
|
| 24 |
+
from tools.rag_tools import search_groundx_for_rag_context
|
| 25 |
+
from tools.multi_agent_workflow_for_research import run_research_workflow
|
| 26 |
+
|
| 27 |
+
def search_knowledge_base_for_context(query: str) -> str:
|
| 28 |
+
"""
|
| 29 |
+
Searches and retrieves relevant context from a knowledge base (GroundX),
|
| 30 |
+
based on the user's query.
|
| 31 |
+
|
| 32 |
+
Example queries:
|
| 33 |
+
- "What are the main features of fuel system of SU-35."
|
| 34 |
+
- "What are the combat potential of SU-35."
|
| 35 |
+
|
| 36 |
+
Args:
|
| 37 |
+
query: The search query supplied by the user.
|
| 38 |
+
|
| 39 |
+
Returns:
|
| 40 |
+
str: Relevant text content that can be used by the LLM to answer the query.
|
| 41 |
+
"""
|
| 42 |
+
logger.info(f"Searching document for RAG context {query}")
|
| 43 |
+
response = search_groundx_for_rag_context(query)
|
| 44 |
+
logger.info(f"RAG Response: {response}")
|
| 45 |
+
return response
|
| 46 |
+
|
| 47 |
+
def research_write_review_topic(query: str) -> str:
|
| 48 |
+
"""
|
| 49 |
+
Helps with writing a report with research, writing, and review on any topic.
|
| 50 |
+
Returns a reviewed topic.
|
| 51 |
+
|
| 52 |
+
The query is a string that contains the topic to be researched and reviewed.
|
| 53 |
+
|
| 54 |
+
Example queries:
|
| 55 |
+
- "Write me a report on the history of the internet."
|
| 56 |
+
- "Write me a report on origin of the universe."
|
| 57 |
+
- "Write me a report on the impact of climate change on polar bears."
|
| 58 |
+
- "Write me a report on the benefits of meditation."
|
| 59 |
+
- "Write me a report on the future of artificial intelligence."
|
| 60 |
+
- "Write me a report on the effects of social media on mental health."
|
| 61 |
+
|
| 62 |
+
Args:
|
| 63 |
+
query (str): The query to research, write and review .
|
| 64 |
+
|
| 65 |
+
Returns:
|
| 66 |
+
str: A nicely formatted string.
|
| 67 |
+
"""
|
| 68 |
+
try:
|
| 69 |
+
logger.info(f"Researching the topic: {query}")
|
| 70 |
+
result = run_research_workflow(query)
|
| 71 |
+
return result or "Research completed, but no content was generated."
|
| 72 |
+
except Exception as e:
|
| 73 |
+
return f"Error: {e}"
|
| 74 |
+
|
| 75 |
+
with gr.Blocks() as server_info:
|
| 76 |
+
gr.Markdown("""
|
| 77 |
+
# MCP powered RAG and Research
|
| 78 |
+
|
| 79 |
+
I present to you a MCP powered RAG and Research.
|
| 80 |
+
|
| 81 |
+
RAG Tool uses GroundX service to fetch the knowledge base. The knowledge base is a document that contains information about the SU-35 aircraft, including its features, capabilities, and specifications.
|
| 82 |
+
Please check [this PDF](https://airgroup2000.com/gallery/albums/userpics/32438/SU-35_TM_eng.pdf) to formulate queries on Sukhoi.
|
| 83 |
+
|
| 84 |
+
The Research Tool is implemented using multi-agent workflow using LlamaIndex (ResearchAgent, WriteAgent, and ReviewAgent).
|
| 85 |
+
|
| 86 |
+
## Available Tools
|
| 87 |
+
|
| 88 |
+
### search_knowledge_base_for_context
|
| 89 |
+
- **Description**: Searches and retrieves relevant context from a knowledge base based on the user's query.
|
| 90 |
+
- **Example Queries**:
|
| 91 |
+
- "What are the main features of fuel system of SU-35."
|
| 92 |
+
- "What are the combat potential of SU-35."
|
| 93 |
+
|
| 94 |
+
### research_write_review_topic
|
| 95 |
+
- **Description**: Helps with writing a report with research, writing, and review on any topic.
|
| 96 |
+
- **Example Queries**:
|
| 97 |
+
- "Write me a report on the history of the internet."
|
| 98 |
+
- "Write me a report on origin of the universe."
|
| 99 |
+
- "Write me a report on the impact of climate change on polar bears."
|
| 100 |
+
|
| 101 |
+
## How to Use
|
| 102 |
+
- Use the MCP RAG Tool tab above to query the knowledge base.
|
| 103 |
+
- Use the Research Tool tab above to write report on any topic.
|
| 104 |
+
|
| 105 |
+
## Demo Link
|
| 106 |
+
[Link to Demo on Youtube](https://www.youtube.com/mcp-rag-research)
|
| 107 |
+
""")
|
| 108 |
+
|
| 109 |
+
mcp_rag_tool = gr.Interface(
|
| 110 |
+
fn=search_knowledge_base_for_context,
|
| 111 |
+
inputs=["text"],
|
| 112 |
+
outputs=[gr.Textbox(label="Knowledge Base", max_lines=10)],
|
| 113 |
+
title="MCP RAG Tool",
|
| 114 |
+
description="Searches and retrieves relevant context from a knowledge base"
|
| 115 |
+
)
|
| 116 |
+
|
| 117 |
+
research_tool = gr.Interface(
|
| 118 |
+
fn=research_write_review_topic,
|
| 119 |
+
inputs=["text"],
|
| 120 |
+
outputs=[gr.Textbox(label="Reviewed Topic", max_lines=10)],
|
| 121 |
+
title="Research Tool",
|
| 122 |
+
description="Helps with report writing with research, writing, and review agents on any topic. ",
|
| 123 |
+
concurrency_limit=10
|
| 124 |
+
)
|
| 125 |
+
|
| 126 |
+
named_interfaces = {
|
| 127 |
+
"Project Information": server_info,
|
| 128 |
+
"RAG - Tool": mcp_rag_tool,
|
| 129 |
+
"Research a Topic - Tool": research_tool
|
| 130 |
+
}
|
| 131 |
+
|
| 132 |
+
# Tab names and interfaces
|
| 133 |
+
tab_names = list(named_interfaces.keys())
|
| 134 |
+
interface_list = list(named_interfaces.values())
|
| 135 |
+
|
| 136 |
+
mcp_server = gr.TabbedInterface(
|
| 137 |
+
interface_list,
|
| 138 |
+
tab_names=tab_names,
|
| 139 |
+
title="π MCP powered RAG and Research π"
|
| 140 |
+
)
|
| 141 |
+
|
| 142 |
+
# Launch the MCP Server
|
| 143 |
+
if __name__ == "__main__":
|
| 144 |
+
mcp_server.queue(default_concurrency_limit=10)
|
| 145 |
+
mcp_server.launch(
|
| 146 |
+
server_name="0.0.0.0",
|
| 147 |
+
server_port=7860,
|
| 148 |
+
share=False,
|
| 149 |
+
debug=False,
|
| 150 |
+
mcp_server=True
|
| 151 |
+
)
|