stwhite
/
ira
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Update Gradio interface to handle async methods

This commit is contained in:
Steve White 2025-02-28 17:08:13 -06:00
parent 3b75436423
commit 4753e567ad
1 changed files with 13 additions and 12 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

25
ui/gradio_interface.py
View File

@ -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]
)