removed debug print statement

app.py

@@ -326,6 +326,7 @@ sidebar()
 header()
 chat_box()
 
+# Main chat loop
 if prompt := st.chat_input("Generate Ebook"):
     st.chat_message("user").markdown(prompt)
     st.session_state.messages.append({"role": "user", "content": prompt})

utils.py

@@ -22,16 +22,16 @@ bingSearchClient = SearchClient("bing", api_key=BING_SEARCH_API_KEY, engine_id=N
 
 
 def rerank(query, top_k, search_results, chunk_size):
-    """Chunks and reranks the documents using specified reranker,
+    """Chunks and reranks the documents using a specified reranker,
 
     Args:
-        query (str): your query
-        top_k (str): _description_
-        search_results ([]{link:str, text:str}): _description_
-        chunk_size (_type_): _description_
+        query (str): the query for reranking
+        top_k (int): the number of top reranked results to return
+        search_results (list[dict]): a list of dictionaries containing "link" and "text" keys for each search result
+        chunk_size (int): the size of each chunk for reranking
 
     Returns:
-        _type_: []{link:str, text:str}
+        list[dict]: a list of dictionaries containing the top reranked results with "link" and "text" keys
     """
     chunks = []
     for result in search_results:
@@ -114,7 +114,7 @@ def gen_augmented_prompt_via_websearch(
     # remove duplicate links
     links = list(set(links))
 
-    print(reranked_results)
+    # print(reranked_results)
 
     prev_output = prev_output if pass_prev else ""