Update main.py

main.py

@@ -72,7 +72,11 @@ async def generate_report(request: Request, query: QueryModel):
         try:
             urls,optimized_search_query = search_brave(search_query, num_results=10)
             # Fetch and extract content from the URLs
-            all_text_with_urls = fetch_and_extract_content(data_format, urls, query_str)
+            all_text_with_urls = fetch_and_extract_content(            
+                                        data_format=data_format,
+                                        urls=urls,
+                                        query=query,
+                                        num_refrences=8)
             # Prepare the prompt for generating the report
             additional_context = limit_tokens(str(all_text_with_urls))
             prompt = f"#### COMPLETE THE TASK: {description} #### IN THE CONTEXT OF ### CONTEXT: {query_str} USING THE #### SCRAPED DATA:{additional_context}"