Update app.py

app.py

@@ -13,8 +13,8 @@ from huggingface_hub import snapshot_download, hf_hub_download
 
 # from prompts import CONDENSE_QUESTION_PROMPT
 
-repo_name = "IlyaGusev/saiga2_7b_gguf"
-model_name = "model-q2_K.gguf"
+repo_name = "IlyaGusev/saiga_mistral_7b_gguf"
+model_name = "model-q4_K.gguf"
 
 #snapshot_download(repo_id=repo_name, local_dir=".", allow_patterns=model_name)
 
@@ -81,22 +81,22 @@ def handle_userinput(user_question):
 
     st.session_state.retrieved_text = response['source_documents']
 
-    for i, message in enumerate(st.session_state.chat_history):
-        if i % 2 == 0:
+    for i, (message, text) in enumerate(zip(st.session_state.chat_history, st.session_state.retrieved_text)):
+        if i % 3 == 0:
             st.write(user_template.replace(
                 "{{MSG}}", message.content), unsafe_allow_html=True)
         else:
             st.write(bot_template.replace(
                 "{{MSG}}", message.content), unsafe_allow_html=True)
-            #st.write(bot_template.replace(
-            #    "{{MSG}}", str(text[0])), unsafe_allow_html=True)
+            st.write(bot_template.replace(
+                "{{MSG}}", str(text[1].page_content])), unsafe_allow_html=True)
 
             
             
-    for n in enumerate(st.session_state.retrieved_text):
-        st.write(response['source_documents'][n])
+    #for text in enumerate(st.session_state.retrieved_text):
+    #    st.write(text[1].page_content, '\n')
 
-    print(response['source_documents'][0])
+    #print(response['source_documents'][0])
 
 # main code
 load_dotenv()