1st commit

.gitignore

@@ -0,0 +1,12 @@
+# Python-generated files
+__pycache__/
+*.py[oc]
+build/
+dist/
+wheels/
+*.egg-info
+
+# Virtual environments
+.venv
+*.lock
+.env

.pre-commit-config.yaml

@@ -0,0 +1,23 @@
+repos:
+  - repo: https://github.com/pre-commit/pre-commit-hooks
+    rev: v4.5.0
+    hooks:
+    - id: check-case-conflict
+    - id: check-merge-conflict
+    - id: check-symlinks
+    - id: check-yaml
+    - id: debug-statements
+    - id: end-of-file-fixer
+    - id: mixed-line-ending
+    - id: requirements-txt-fixer
+    - id: trailing-whitespace
+
+  - repo: https://github.com/astral-sh/ruff-pre-commit
+    rev: v0.4.1
+    hooks:
+      # Run the linter.
+      - id: ruff
+        args: [ --fix, "--ignore", "F821,F401,F811"]
+      # Run the formatter.
+      - id: ruff-format
+        types_or: [python, jupyter]

.python-version

@@ -0,0 +1 @@
+3.12

agent.py

@@ -0,0 +1,27 @@
+import logfire
+from deep_research_agent.nodes import (
+    BeginResearch,
+    FinalReport,
+    Researcher,
+    Supervisor,
+    WriteResearchBrief,
+)
+from deep_research_agent.state import ResearchState
+from pydantic_graph import Graph
+
+logfire.configure()
+logfire.instrument_pydantic_ai()
+
+
+async def query_agent(research_query: str):
+    state = ResearchState()
+    graph = Graph(
+        nodes=(BeginResearch, WriteResearchBrief, Supervisor, Researcher, FinalReport)
+    )
+
+    result = await graph.run(
+        start_node=BeginResearch(query=research_query),
+        state=state,
+    )
+
+    return result.output

app.py

@@ -0,0 +1,62 @@
+import gradio as gr
+import asyncio
+from agent import query_agent
+
+
+# Simulate your async agent interaction
+async def interact_with_agent(prompt: str):
+    # # This is where you'd call your real query_agent()
+    # await asyncio.sleep(1)
+    # urls = [
+    #     "https://arxiv.org/abs/1234.5678",
+    #     "https://openai.com/research",
+    #     "https://example.com/article",
+    # ]
+    # report = f"Report generated for: '{prompt}'"
+    final_report = await query_agent(research_query=prompt)
+    final_report, refs = (
+        final_report.split("\n\n## References:\n\n")[0],
+        final_report.split("\n\n## References:\n\n")[1],
+    )
+    return final_report, refs
+
+
+# Display URLs in Markdown
+def format_urls(urls):
+    if not urls:
+        return "(No sources yet)"
+    return urls
+
+
+with gr.Blocks(title="Deep Research Agent") as chat_app:
+    gr.Markdown("## 🧠 Deep Research Agent")
+
+    with gr.Row():
+        # Main chat area
+        with gr.Column(scale=4):
+            chatbot = gr.Chatbot(type="messages", label="Chat", height=550)
+            msg = gr.Textbox(placeholder="Ask a research question...")
+            clear = gr.ClearButton([msg, chatbot])
+
+        # Sidebar for URLs
+        with gr.Column(scale=1):
+            gr.Markdown("### 🔗 References")
+            url_display = gr.Markdown("(No sources yet)")
+
+    # Step 1: show user message immediately
+    def show_user_msg(message, chat_history: list):
+        chat_history.append({"role": "user", "content": message})
+        return chat_history
+
+    # Step 2: respond asynchronously (simulate the agent)
+    async def respond(message: str, chat_history: list):
+        report, urls = await interact_with_agent(message)
+        chat_history.append({"role": "assistant", "content": report})
+        return "", chat_history, format_urls(urls)
+
+    # Chain both: show user → respond with URLs
+    msg.submit(show_user_msg, [msg, chatbot], [chatbot], queue=False).then(
+        respond, [msg, chatbot], [msg, chatbot, url_display]
+    )
+
+chat_app.launch()

pyproject.toml

@@ -0,0 +1,18 @@
+[project]
+name = "dra-app"
+version = "0.1.0"
+description = "Add your description here"
+readme = "README.md"
+requires-python = ">=3.12"
+dependencies = [
+    "gradio>=5.49.1",
+    "pydantic-deep-research-agent",
+]
+
+[dependency-groups]
+dev = [
+    "pre-commit>=4.3.0",
+]
+
+[tool.uv.sources]
+pydantic-deep-research-agent = { git = "https://github.com/alexliap/pydantic_deep_research_agent" }

requirements.txt

@@ -0,0 +1,2 @@
+gradio>=5.49.1
+pydantic-deep-research-agent @ git+https://github.com/alexliap/pydantic_deep_research_agent