Update app.py

app.py

@@ -324,6 +324,7 @@ def search_glossary(query):
     # 🕵️‍♂️ Searching the glossary for: query
     all_results = ""
     #st.markdown(f"- {query}")
+
     
     # 🔍 ArXiv RAG researcher expert ~-<>-~ Paper Summary & Ask LLM
     client = Client("awacke1/Arxiv-Paper-Search-And-QA-RAG-Pattern")
@@ -334,9 +335,11 @@ def search_glossary(query):
         stream_outputs=True,
         api_name="/ask_llm"
     )
+    st.markdown("# Mixtral-8x7B-Instruct-v0.1")
     st.markdown(result)
-    
     st.code(result, language="python", line_numbers=True)
+
+    
     # 🔍 ArXiv RAG researcher expert ~-<>-~ Paper Summary & Ask LLM - api_name: /ask_llm
     result2 = client.predict(
         prompt=query,
@@ -344,9 +347,11 @@ def search_glossary(query):
         stream_outputs=True,
         api_name="/ask_llm"
     )
+    st.markdown("# Mistral-7B-Instruct-v0.2")
     st.markdown(result2)
-    
     st.code(result2, language="python", line_numbers=True)
+
+    
     # 🔍 ArXiv RAG researcher expert ~-<>-~ Paper Summary & Ask LLM - api_name: /ask_llm
     result3 = client.predict(
         prompt=query,
@@ -354,9 +359,11 @@ def search_glossary(query):
         stream_outputs=True,
         api_name="/ask_llm"
     )
+    st.markdown("# Gemma-7b-it")
     st.markdown(result3)
-    
     st.code(result3, language="python", line_numbers=True)
+
+    
     # 🔍 ArXiv RAG researcher expert ~-<>-~ Paper Summary & Ask LLM - api_name: /update_with_rag_md
     response2 = client.predict(
         message=query,	# str  in 'parameter_13' Textbox component
@@ -365,9 +372,11 @@ def search_glossary(query):
         llm_model_picked="mistralai/Mistral-7B-Instruct-v0.2",
         api_name="/update_with_rag_md"
     )
+    st.markdown("# Mistral-7B-Instruct-v0.2 update_with_rag_md 0")
     st.markdown(response2[0])
     st.code(response2[0], language="python", line_numbers=True, wrap_lines=True) 
     
+    st.markdown("# Mistral-7B-Instruct-v0.2 update_with_rag_md 1")
     st.markdown(response2[1])
     st.code(response2[1], language="python", line_numbers=True, wrap_lines=True)
 
@@ -386,15 +395,15 @@ def search_glossary(query):
     except:
         st.markdown('2 error')        
     try:
-        filename = generate_filename(result3, "md")
-        create_file(filename, query, result3)
-        st.session_state.chat_history.append({"assistant": query, "ArXiV": result3})
+        filename = generate_filename(response2[0], "md")
+        create_file(filename, query, response2[0])
+        st.session_state.chat_history.append({"assistant": query, "ArXiV": response2[0]})
     except:
         st.markdown('3 error')            
     try:
-        filename = generate_filename(response2, "md")
-        create_file(filename, query, response2)
-        st.session_state.chat_history.append({"assistant": query, "ArXiV": response2})
+        filename = generate_filename(response2[1], "md")
+        create_file(filename, query, response2[1])
+        st.session_state.chat_history.append({"assistant": query, "ArXiV": response2[1]})
     except:
         st.markdown('4 error')