Limit communities to 2, Add default value to search strategy

app.py

@@ -58,7 +58,9 @@ demo = gr.ChatInterface(
             value="You are a medical assistant Chatbot. For any query that you don't know, you will say 'I don't know'. You will answer with the given information:",
             label="System message",
         ),
-        gr.Dropdown(choices=["Local", "Global"], label="Select search strategy"),
+        gr.Dropdown(
+            choices=["Local", "Global"], value="Local", label="Select search strategy"
+        ),
         gr.Slider(
             minimum=0.1,
             maximum=1.0,

rag.py

@@ -222,12 +222,10 @@ def global_retriever(query: str, level: int, response_type: str):
 
 
     ---Target response length and format---
-
     {response_type}
 
 
     ---Analyst Reports---
-
     {report_data}
 
     Add sections and commentary to the response as appropriate for the length and format. Do not add references in your answer.
@@ -268,13 +266,13 @@ def global_retriever(query: str, level: int, response_type: str):
     # print(community_data)
     intermediate_results = []
     i = 0
-    for community in tqdm(community_data[:5], desc="Processing communities"):
+    for community in tqdm(community_data[:2], desc="Processing communities"):
         intermediate_response = map_chain.invoke(
             {"question": query, "context_data": community["output"]}
         )
         intermediate_results.append(intermediate_response)
         i += 1
-
+    print(intermediate_results)
     ###Debug####
     # tokens = global_tokenizer(intermediate_results)
     # print(f"Number of input tokens: {len(tokens)}")