Update main.py

main.py

@@ -9,6 +9,7 @@ from datetime import datetime, timedelta
 from fastapi_cache import FastAPICache
 from fastapi_cache.backends.inmemory import InMemoryBackend
 from fastapi_cache.decorator import cache
+import asyncio
 
 # Load environment variables from .env file
 #load_dotenv("keys.env")
@@ -77,9 +78,9 @@ async def generate_report(query: QueryModel):
     all_text_with_urls = [("", "")]
 
     if query.internet:
-        search_query = description
+        search_query = re.sub(r'[^\w\s]', '', description).strip()
         try:
-            urls, optimized_search_query = search_brave(search_query, num_results=4)
+            urls, optimized_search_query = await search_brave(search_query, num_results=4)
             all_text_with_urls = fetch_and_extract_content(data_format, urls, query_str)
             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}"