Update app.py

app.py

@@ -63,8 +63,8 @@ def load_file(file_name):
         content = file.read()
     return content
 
-def main():
-    st.title("Memory Flag System")
+def MemoryFlagSystem(query):
+    st.title("▶️ Memory Flag System")
 
     client = Client("awacke1/Arxiv-Paper-Search-And-QA-RAG-Pattern")
 
@@ -73,7 +73,7 @@ def main():
     search_source = st.selectbox("Search Source", ["Semantic Search - up to 10 Mar 2024", "Arxiv Search - Latest - (EXPERIMENTAL)"])
     llm_model = st.selectbox("LLM Model", ["mistralai/Mixtral-8x7B-Instruct-v0.1", "mistralai/Mistral-7B-Instruct-v0.2", "google/gemma-7b-it", "None"])
 
-    if st.button("Search"):
+    if st.button("🔎 Search"):
         result = client.predict(
             search_query,
             top_n_results,
@@ -1216,6 +1216,9 @@ def StreamMedChatResponse(topic):
 def arxivmain(query):
     prompt = query
     with st.expander("Prompts 📚", expanded=True):
+
+        MemoryFlagSystem(query)
+
         example_input = st.text_input("Enter your prompt text:", value=prompt, help="Enter text to get a response.")
         if st.button("Run Prompt", help="Click to run."):
             try:
@@ -1385,7 +1388,6 @@ if st.button("Clear Query Parameters", key='ClearQueryParams'):
 # 18. Run AI Pipeline
 if __name__ == "__main__":
     whisper_main()
-    main()