Update main.py

main.py

@@ -62,13 +62,15 @@ async def generate_report(request: Request, query: QueryModel):
     internet = "online" if query.internet else "offline"
     sys_prompt_output_format = sys_prompts[internet][query.output_format]
     data_format = query.data_format
-
+    optimized_search_query = ""
+    all_text_with_urls = [("","")]
+    
     # Combine query with user keywords
     if query.internet:
         search_query = description
         # Search for relevant URLs
         try:
-            urls = search_brave(search_query, num_results=4)
+            urls,optimized_search_query = search_brave(search_query, num_results=4)
             # Fetch and extract content from the URLs
             all_text_with_urls = fetch_and_extract_content(data_format, urls, query_str)
             # Prepare the prompt for generating the report
@@ -81,7 +83,6 @@ async def generate_report(request: Request, query: QueryModel):
     
     if not query.internet:
         prompt = f"#### COMPLETE THE TASK: {description} #### IN THE CONTEXT OF ### CONTEXT: {query_str}"
-        all_text_with_urls = [("","")]
 
     md_report = together_response(prompt, model=llm_default_medium, SysPrompt=sys_prompt_output_format)
     
@@ -95,7 +96,8 @@ async def generate_report(request: Request, query: QueryModel):
     # Return the generated report
     return {
         "report": md_to_html(md_report),
-        "references": references_html
+        "references": references_html,
+        "search_qury":optimized_search_query
     }
 
 app.add_middleware(