Spaces:
Runtime error
Runtime error
Create app.py
Browse files
app.py
ADDED
|
@@ -0,0 +1,147 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
import os
|
| 2 |
+
import gradio as gr
|
| 3 |
+
from duckduckgo_search import DDGS
|
| 4 |
+
from gradio_client import Client
|
| 5 |
+
import time
|
| 6 |
+
|
| 7 |
+
def generate_search_queries(topic):
|
| 8 |
+
"""Generate optimized search queries."""
|
| 9 |
+
prompt = f"""
|
| 10 |
+
<Instructions>
|
| 11 |
+
You are an AI research strategist that generates optimized search queries for investigating complex topics. When I provide a <topic>, create 10-15 search terms/phrases that would effectively discover relevant information through search engines and academic databases.
|
| 12 |
+
|
| 13 |
+
Rules for query generation:
|
| 14 |
+
1. Include 3 levels of specificity: broad conceptual terms, mid-range topic phrases, niche technical terms
|
| 15 |
+
2. Cover multiple research angles: definitions, controversies, applications, case studies, trends
|
| 16 |
+
3. Use both quoted exact-match phrases and natural language questions
|
| 17 |
+
4. Include synonyms and variant terminology
|
| 18 |
+
5. Avoid duplicate concepts - each query must target distinct information
|
| 19 |
+
6. Order queries from general to specific
|
| 20 |
+
|
| 21 |
+
Example response format:
|
| 22 |
+
<search_queries>
|
| 23 |
+
<query>[1] "generative AI" AND intellectual property</query>
|
| 24 |
+
<query>[2] Training data sourcing legality LLM</query>
|
| 25 |
+
...
|
| 26 |
+
</search_queries>
|
| 27 |
+
|
| 28 |
+
Now process this topic:
|
| 29 |
+
<topic>{topic}</topic>
|
| 30 |
+
</Instructions>
|
| 31 |
+
"""
|
| 32 |
+
try:
|
| 33 |
+
response = DDGS().chat(prompt, model='o3-mini')
|
| 34 |
+
queries = []
|
| 35 |
+
|
| 36 |
+
for part in response.split("</query>"):
|
| 37 |
+
if "<query>" in part:
|
| 38 |
+
query_text = part.split("<query>")[-1].strip()
|
| 39 |
+
if query_text:
|
| 40 |
+
clean_query = query_text.split("] ", 1)[-1] if "] " in query_text else query_text
|
| 41 |
+
queries.append(clean_query)
|
| 42 |
+
|
| 43 |
+
if not queries:
|
| 44 |
+
return [f"{topic} historical analysis",
|
| 45 |
+
f"{topic} primary sources",
|
| 46 |
+
f"{topic} geopolitical impact"]
|
| 47 |
+
|
| 48 |
+
return queries[:15]
|
| 49 |
+
|
| 50 |
+
except Exception as e:
|
| 51 |
+
print(f"Error generating queries: {str(e)}")
|
| 52 |
+
return [topic]
|
| 53 |
+
|
| 54 |
+
def conduct_research(query):
|
| 55 |
+
"""Conduct deep research on a single query"""
|
| 56 |
+
client = Client("m-ric/open_Deep-Research")
|
| 57 |
+
client.predict(query, api_name="/log_user_message")
|
| 58 |
+
research_data = client.predict([], api_name="/interact_with_agent")
|
| 59 |
+
|
| 60 |
+
for msg in reversed(research_data):
|
| 61 |
+
if "Final answer:" in msg['content']:
|
| 62 |
+
return msg['content'].split("Final answer:")[-1].strip()
|
| 63 |
+
return "No conclusive information found"
|
| 64 |
+
|
| 65 |
+
def synthesize_results(original_query, queries, findings):
|
| 66 |
+
"""Synthesize research findings into final summary"""
|
| 67 |
+
synthesis_prompt = f"""
|
| 68 |
+
<Inputs>
|
| 69 |
+
Original Query: {original_query}
|
| 70 |
+
Research Queries: {queries}
|
| 71 |
+
Research Findings: {findings}
|
| 72 |
+
</Inputs>
|
| 73 |
+
|
| 74 |
+
<Instructions>
|
| 75 |
+
You are an analytical research synthesizer. Merge these findings into one cohesive summary:
|
| 76 |
+
|
| 77 |
+
1. Start with 1 paragraph overview
|
| 78 |
+
2. Bullet points of key findings (minimum 5)
|
| 79 |
+
3. 1 paragraph synthesis connecting findings to original query
|
| 80 |
+
4. "Additional Notes" section for peripheral but useful details
|
| 81 |
+
|
| 82 |
+
Rules:
|
| 83 |
+
- Include EVERY relevant data point
|
| 84 |
+
- Natural conversational English but professional
|
| 85 |
+
- No markdown formatting
|
| 86 |
+
- Keep paragraphs under 5 sentences
|
| 87 |
+
|
| 88 |
+
Example structure:
|
| 89 |
+
"Three separate analyses concur... [specific data]... This suggests... [connection to query]..."
|
| 90 |
+
|
| 91 |
+
Begin by confirming understanding of the core query, then proceed with synthesis.
|
| 92 |
+
</Instructions>
|
| 93 |
+
"""
|
| 94 |
+
|
| 95 |
+
synthesizer = Client("MiniMaxAI/MiniMax-Text-01")
|
| 96 |
+
return synthesizer.predict(
|
| 97 |
+
message=synthesis_prompt,
|
| 98 |
+
max_tokens=1000,
|
| 99 |
+
temperature=0.1,
|
| 100 |
+
top_p=0.9,
|
| 101 |
+
api_name="/chat"
|
| 102 |
+
)
|
| 103 |
+
|
| 104 |
+
def deep_research_agent(topic):
|
| 105 |
+
queries = generate_search_queries(topic)
|
| 106 |
+
print(f"🔍 Generated {len(queries)} research queries")
|
| 107 |
+
|
| 108 |
+
findings = []
|
| 109 |
+
for i, query in enumerate(queries, 1):
|
| 110 |
+
print(f"⏳ Researching query {i}/{len(queries)}: {query}")
|
| 111 |
+
findings.append(conduct_research(query))
|
| 112 |
+
time.sleep(1)
|
| 113 |
+
|
| 114 |
+
print("🧠 Synthesizing findings...")
|
| 115 |
+
return synthesize_results(topic, queries, findings)
|
| 116 |
+
|
| 117 |
+
def create_interface():
|
| 118 |
+
with gr.Blocks(analytics_enabled=False) as app:
|
| 119 |
+
gr.Markdown("# Stealth Research Assistant")
|
| 120 |
+
with gr.Row():
|
| 121 |
+
topic_input = gr.Textbox(label="Research Topic", max_lines=1)
|
| 122 |
+
submit_btn = gr.Button("Start Analysis", variant="primary")
|
| 123 |
+
|
| 124 |
+
status = gr.Textbox(label="Operation Status", value="Ready", interactive=False)
|
| 125 |
+
output = gr.Textbox(label="Final Report", lines=15, interactive=False)
|
| 126 |
+
|
| 127 |
+
@submit_btn.click(inputs=topic_input, outputs=[output, status], api_name=False)
|
| 128 |
+
def execute_analysis(topic):
|
| 129 |
+
try:
|
| 130 |
+
yield ["", "Analyzing topic..."]
|
| 131 |
+
result = deep_research_agent(topic)
|
| 132 |
+
yield [result, "Completed"]
|
| 133 |
+
except Exception as e:
|
| 134 |
+
yield ["", f"Error: {str(e)[:200]}"]
|
| 135 |
+
|
| 136 |
+
return app
|
| 137 |
+
|
| 138 |
+
def launch():
|
| 139 |
+
interface = create_interface()
|
| 140 |
+
interface.queue().launch(
|
| 141 |
+
server_name=os.getenv("SERVER_HOST", "127.0.0.1"),
|
| 142 |
+
server_port=int(os.getenv("SERVER_PORT", "7860")),
|
| 143 |
+
show_api=False
|
| 144 |
+
)
|
| 145 |
+
|
| 146 |
+
if __name__ == "__main__":
|
| 147 |
+
launch()
|