Spaces:
Running
Running
| # SPDX-License-Identifier: MIT | |
| import json | |
| from pathlib import Path | |
| import gradio as gr | |
| from graphrag_audit.communities import LeidenDetector | |
| from graphrag_audit.eval import run_evaluation | |
| from graphrag_audit.extract import extract_repository | |
| from graphrag_audit.graph import KnowledgeGraph | |
| from graphrag_audit.query import GraphRAG | |
| from graphrag_audit.summarize import summarize_communities | |
| ROOT = Path(__file__).resolve().parent / "sample_repo" | |
| CORPUS = extract_repository(ROOT) | |
| GRAPH = KnowledgeGraph().from_corpus(CORPUS) | |
| HIERARCHY = LeidenDetector().detect(GRAPH) | |
| REPORTS = summarize_communities(HIERARCHY, CORPUS) | |
| RAG = GraphRAG(CORPUS, GRAPH, REPORTS) | |
| METRICS = run_evaluation(str(ROOT), n=100) | |
| def ask(question, mode): | |
| result = RAG.query(question, mode=mode) | |
| return json.dumps({ | |
| "answer": result.answer, | |
| "citations": result.citations, | |
| "communities": result.communities, | |
| "method": HIERARCHY.method, | |
| "communities_total": len(HIERARCHY.communities), | |
| "eval": {"graph": METRICS["graph"], "flat": METRICS["flat"], "win_rate": METRICS["win_rate"]}, | |
| }, indent=2) | |
| with gr.Blocks(title="GraphRAG CodeAudit Explorer") as demo: | |
| gr.Markdown("# GraphRAG CodeAudit Explorer\nAsk a local or global question over the synthetic sample repository.") | |
| question = gr.Textbox(label="Question", value="What security risks exist in auth.py?") | |
| mode = gr.Radio(["global", "local"], value="global", label="Search mode") | |
| button = gr.Button("Ask") | |
| output = gr.Code(label="GraphRAG response", language="json") | |
| button.click(ask, inputs=[question, mode], outputs=output) | |
| demo.load(ask, inputs=[question, mode], outputs=output) | |
| if __name__ == "__main__": | |
| demo.launch() | |