fixing sourcereturn and upping sources to 5

app.py

@@ -25,7 +25,7 @@ client = OpenAIOG()
 # Load index for retrieval
 storage_context = StorageContext.from_defaults(persist_dir="arv_metadata")
 index = load_index_from_storage(storage_context)
-retriever = index.as_retriever(similarity_top_k=3)
+retriever = index.as_retriever(similarity_top_k=5)
 
 # Define keyword lists
 acknowledgment_keywords_sw = ["sawa", "ndiyo", "naam", "hakika", "asante", "nimeelewa", "nimekupata", "ni kweli", "kwa hakika", "nimesikia", "ahsante"]
@@ -96,10 +96,12 @@ def nishauri(question, conversation_history: list[str]):
         )
         reply_to_user = completion.choices[0].message.content
         conversation_history.append({"user": question, "chatbot": reply_to_user})
-        Source1Return = ""
-        Source2Return = ""
-        Source3Return = ""
-        return reply_to_user, source1return, source2return, source3return, conversation_history
+        source1Return = ""
+        source2Return = ""
+        source3Return = ""
+        source4Return = ""
+        source5Return = ""
+        return reply_to_user, source1return, source2return, source3return, source4return, source5return, conversation_history
 
     # Detect language and translate if needed
     lang_question = detect_language(question)
@@ -108,7 +110,7 @@ def nishauri(question, conversation_history: list[str]):
     
     # Retrieve relevant sources
     sources = retriever.retrieve(question)
-    retrieved_text = "\n\n".join([f"Source {i+1}: {source.text}" for i, source in enumerate(sources[:3])])
+    retrieved_text = "\n\n".join([f"Source {i+1}: {source.text}" for i, source in enumerate(sources[:5])])
 
     source1return = ("File Name: " +
                      sources[0].metadata["file_name"] +
@@ -131,6 +133,20 @@ def nishauri(question, conversation_history: list[str]):
                      "\n Source Text: " +
                      sources[2].text)
 
+    source4return = ("File Name: " +
+                     sources[3].metadata["file_name"] +
+                     "\nPage Number: " +
+                     sources[3].metadata["page_label"] +
+                     "\n Source Text: " +
+                     sources[3].text)
+
+    source5return = ("File Name: " +
+                     sources[4].metadata["file_name"] +
+                     "\nPage Number: " +
+                     sources[4].metadata["page_label"] +
+                     "\n Source Text: " +
+                     sources[4].text)
+
     # Combine into new user question - conversation history, new question, retrieved sources
     question_final = (
         f"The user asked the following question: \"{question}\"\n\n"
@@ -189,7 +205,7 @@ def nishauri(question, conversation_history: list[str]):
         reply_to_user = GoogleTranslator(source='auto', target='sw').translate(reply_to_user) 
 
     # return system_prompt, conversation_history 
-    return reply_to_user, source1return, source2return, source3return, conversation_history 
+    return reply_to_user, source1return, source2return, source3return, source4return, source5return, conversation_history 
 
 #%%
 demo = gr.Interface(
@@ -202,6 +218,8 @@ demo = gr.Interface(
         gr.Textbox(label = "Source 1", max_lines = 10, autoscroll = False, type = "text"),
         gr.Textbox(label = "Source 2", max_lines = 10, autoscroll = False, type = "text"),
         gr.Textbox(label = "Source 3", max_lines = 10, autoscroll = False, type = "text"),
+        gr.Textbox(label = "Source 4", max_lines = 10, autoscroll = False, type = "text"),
+        gr.Textbox(label = "Source 5", max_lines = 10, autoscroll = False, type = "text"),
         gr.State()
             ],
 )