Update app.py

app.py

@@ -13,7 +13,7 @@ import gradio as gr
 
 # -----------------------------------------------------------------------------
 # Default configuration and Prompts
-NUM_SEARCH = 8 # Number of links to parse from Google
+NUM_SEARCH = 8  # Number of links to parse from Google
 SEARCH_TIME_LIMIT = 7  # Max seconds to request website sources before skipping to the next URL
 TOTAL_TIMEOUT = 25  # Overall timeout for all operations
 MAX_CONTENT = 500  # Number of words to add to LLM context for each search result
@@ -90,13 +90,14 @@ def fetch_webpage(url, timeout):
 
 def parse_google_results(query, num_search=NUM_SEARCH, search_time_limit=SEARCH_TIME_LIMIT):
     """Perform a Google search and parse the content of the top results."""
-    urls = search(query, num_results=num_search)
+    urls = list(search(query, num_results=num_search))
     max_workers = os.cpu_count() or 1  # Fallback to 1 if os.cpu_count() returns None
     with ThreadPoolExecutor(max_workers=max_workers) as executor:
         future_to_url = {executor.submit(fetch_webpage, url, search_time_limit): url for url in urls}
         return {url: page_text for future in as_completed(future_to_url) if (url := future.result()[0]) and (page_text := future.result()[1])}
 
 def save_markdown(content, file_path):
+    """Save content to a Markdown file."""
     with open(file_path, 'a') as file:
         file.write(content)
 
@@ -112,7 +113,7 @@ def llm_check_search(query, file_path, msg_history=None, llm_model=LLM_MODEL):
         max_tokens=30
     ).choices[0].message.content
 
-    # check if the response contains "ns"
+    # Check if the response contains "ns"
     cleaned_response = response.lower().strip()
     if re.fullmatch(r"\bns\b", cleaned_response):
         print("No Google search required.")
@@ -157,7 +158,7 @@ def llm_answer(query, file_path, msg_history=None, search_dic=None, llm_model=LL
             save_markdown(chunk_content, file_path)
     
     print("\n" + "*" * 21 + " LLM END " + "*" * 21 + "\n")
-    # change the line for the next question
+    # Change the line for the next question
     save_markdown("\n\n", file_path)
     new_msg_history = new_msg_history + [{"role": "assistant", "content": ''.join(content)}]
 
@@ -167,7 +168,7 @@ def main_interface(query, file_path="playground.md"):
     """Main function to execute the search, generate response, and save to markdown."""
     msg_history = None
     save_path = None
-    # start with an empty file
+    # Start with an empty file
     with open(file_path, 'w') as file:
         pass