Upload 28 files

src/ui/gradio_app.py

@@ -2,7 +2,8 @@
 
 from __future__ import annotations
 
-from typing import Any, List
+from pathlib import Path
+from typing import Any, List, Optional
 
 import gradio as gr
 
@@ -16,6 +17,8 @@ from src.services.neo4j_service import Neo4jService, Neo4jConnectionError
 class GradioApp:
     """Gradio controller for ingestion and query-time interactions."""
 
+    SAMPLE_REPORTS_DIR = Path(__file__).parent.parent.parent / "sample_project_reports"
+
     TITLE = "Project Intelligence Hub"
     DESCRIPTION = """
 # Project Intelligence Hub
@@ -67,11 +70,34 @@ Transform unstructured PDF reports into a queryable knowledge graph.
             for issue in issues:
                 print(f"  - {issue}")
 
-    def _ingest_action(self, pdf_files: List[Any], clear_db: str):
+    def _get_sample_files(self) -> List[str]:
+        """Return list of available sample PDF files."""
+        if not self.SAMPLE_REPORTS_DIR.exists():
+            return []
+        return sorted([
+            f.name for f in self.SAMPLE_REPORTS_DIR.glob("*.pdf")
+        ])
+
+    def _resolve_sample_files(self, selected_names: Optional[List[str]]) -> List[str]:
+        """Convert selected sample file names to full paths."""
+        if not selected_names:
+            return []
+        return [
+            str(self.SAMPLE_REPORTS_DIR / name)
+            for name in selected_names
+            if (self.SAMPLE_REPORTS_DIR / name).exists()
+        ]
+
+    def _ingest_action(self, pdf_files: List[Any], sample_files: Optional[List[str]], clear_db: str):
         clear_db_bool = clear_db == "Yes"
 
-        if not pdf_files:
-            yield "No documents provided. Upload at least one PDF.", gr.update(value=0, visible=True), None
+        # Combine uploaded files with selected sample files
+        all_files = list(pdf_files) if pdf_files else []
+        sample_paths = self._resolve_sample_files(sample_files)
+        all_files.extend(sample_paths)
+
+        if not all_files:
+            yield "No documents provided. Upload or select at least one PDF.", gr.update(value=0, visible=True), None
             return
 
         if not self.settings.together_ai.api_key:
@@ -100,7 +126,7 @@ Transform unstructured PDF reports into a queryable knowledge graph.
 
             final_state = None
             for status, progress, state in builder.ingest_with_progress(
-                pdf_files=pdf_files,
+                pdf_files=all_files,
                 neo4j_config=neo4j_config,
                 clear_db=clear_db_bool,
                 skip_llm_extraction=True,
@@ -206,11 +232,26 @@ Transform unstructured PDF reports into a queryable knowledge graph.
 
             with gr.Group():
                 pdfs = gr.File(
-                    label="Document Source",
+                    label="Upload Documents",
                     file_types=[".pdf"],
                     file_count="multiple",
                 )
 
+                sample_file_choices = self._get_sample_files()
+                if sample_file_choices:
+                    sample_files = gr.CheckboxGroup(
+                        label="Or Select from Sample Reports",
+                        choices=sample_file_choices,
+                        value=[],
+                    )
+                else:
+                    sample_files = gr.CheckboxGroup(
+                        label="Sample Reports",
+                        choices=[],
+                        value=[],
+                        visible=False,
+                    )
+
                 with gr.Row():
                     clear_toggle = gr.Radio(
                         label="Reset graph before ingestion",
@@ -256,7 +297,7 @@ Transform unstructured PDF reports into a queryable knowledge graph.
 
             ingest_btn.click(
                 fn=self._ingest_action,
-                inputs=[pdfs, clear_toggle],
+                inputs=[pdfs, sample_files, clear_toggle],
                 outputs=[ingest_status, progress_bar, state],
             )