Spaces:
Sleeping
Sleeping
File size: 2,451 Bytes
c049b12 fb1de84 c049b12 fb1de84 c049b12 fb1de84 c049b12 fb1de84 c049b12 fb1de84 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 |
import gradio as gr
from dotenv import load_dotenv
from research_manager import ResearchManager
load_dotenv()
async def ask_clarifications(query: str) -> tuple[str, list[str]]:
"""Generate clarifying questions for *query* and return both a nicely formatted string and the raw list."""
manager = ResearchManager()
questions = await manager.get_clarifying_questions(query)
if not questions:
return (
"No clarifying questions were generated. You can proceed to run the research.",
[],
)
formatted = "\n".join(f"{idx+1}. {q}" for idx, q in enumerate(questions))
return formatted, questions
async def run_research(query: str, answers: str, questions: list[str]):
"""Run the complete research pipeline and stream the markdown report."""
clarifications_block = ""
answer_lines = [line.strip() for line in answers.split("\n")]
while len(answer_lines) < len(questions):
answer_lines.append("")
q_and_a = []
for idx, question in enumerate(questions):
answer = answer_lines[idx]
q_and_a.append(f"Q{idx+1}: {question}\nA{idx+1}: {answer}")
clarifications_block = "\n".join(q_and_a)
async for chunk in ResearchManager().run(query, clarifications_block):
yield chunk
with gr.Blocks(theme=gr.themes.Default(primary_hue="yellow")) as ui:
gr.Markdown("# Deep Research")
with gr.Row():
query_textbox = gr.Textbox(
label="What topic would you like to research?",
placeholder="e.g. How to create a Deep Research Agent?",
)
ask_button = gr.Button("Ask clarifying questions")
clarifying_questions_state = gr.State([])
clarifications_markdown = gr.Markdown(label="Clarifying questions will appear here")
clarification_answers_box = gr.Textbox(
label="Your answers to the clarifying questions (one per line)",
placeholder="Answer 1\nAnswer 2\n...",
lines=3,
)
run_button = gr.Button("Run research", variant="primary")
report = gr.Markdown(label="Report")
ask_button.click(
fn=ask_clarifications,
inputs=query_textbox,
outputs=[clarifications_markdown, clarifying_questions_state],
)
run_button.click(
fn=run_research,
inputs=[query_textbox, clarification_answers_box, clarifying_questions_state],
outputs=report,
)
if __name__ == "__main__":
ui.launch()
|