Update main.py

main.py

@@ -105,13 +105,14 @@ async def generate_report(query: QueryModel):
     data_format = query.data_format
     optimized_search_query = ""
     all_text_with_urls = [("", "")]
+    full_search_object = {}
     generate_charts = query.generate_charts
     output_as_md = query.output_as_md
     
     if query.internet:
         search_query = re.sub(r'[^\w\s]', '', description).strip()
         try:
-            urls, optimized_search_query = search_brave(search_query, num_results=8)
+            urls, optimized_search_query, full_search_object = search_brave(search_query, num_results=8)
             all_text_with_urls = fetch_and_extract_content(data_format, urls, optimized_search_query)
             reference = limit_tokens(str(all_text_with_urls),token_limit=5000)
             user_prompt_final = user_prompt_final.format(description=description, reference=reference)
@@ -156,6 +157,7 @@ async def generate_report(query: QueryModel):
         "report": final_report,
         "references": references_html,
         "search_query": optimized_search_query
+        "search_data_full":full_search_object
     }
 
 @app.post("/generate_report")