diff --git a/ui/gradio_interface.py b/ui/gradio_interface.py index 063c6b0..41db991 100644 --- a/ui/gradio_interface.py +++ b/ui/gradio_interface.py @@ -184,19 +184,20 @@ class GradioInterface: return markdown - async def generate_report(self, query, detail_level, custom_model=None, process_thinking_tags=False, results_file=None): + async def generate_report(self, query, detail_level="standard", custom_model=None, + results_file=None, process_thinking_tags=False): """ - Generate a report from a query. + Generate a report for the given query. Args: - query (str): The query to process - detail_level (str): Detail level for the report - custom_model (str): Custom model to use for report generation - process_thinking_tags (bool): Whether to process thinking tags in the output - results_file (str): Path to results file (optional) + query: The query to generate a report for + detail_level: The level of detail for the report (brief, standard, detailed, comprehensive) + custom_model: Custom model to use for report generation + results_file: Path to a file containing search results + process_thinking_tags: Whether to process thinking tags in the model output Returns: - tuple: (report_markdown, report_file_path) + Path to the generated report """ try: # Create a timestamped output file @@ -239,10 +240,10 @@ class GradioInterface: print(f"No results file provided, performing search for: {query}") # Process the query to create a structured query - structured_query = self.query_processor.process_query(query) + structured_query = await self.query_processor.process_query(query) # Generate search queries for different engines - structured_query = self.query_processor.generate_search_queries( + structured_query = await self.query_processor.generate_search_queries( structured_query, self.search_executor.get_available_search_engines() ) @@ -551,9 +552,9 @@ class GradioInterface: ) report_button.click( - fn=lambda q, d, m, p, f: asyncio.run(self.generate_report(q, d, m, p, f)), + fn=lambda q, d, m, r, p: asyncio.run(self.generate_report(q, d, m, r, p)), inputs=[report_query_input, report_detail_level, report_custom_model, - report_process_thinking, search_file_output], + search_file_output, report_process_thinking], outputs=[report_output, report_file_output] )