Update app.py

app.py

@@ -311,11 +311,20 @@ def archive_current_container(database_name, container_name, client):
     except Exception as e:
         return f"An error occurred while archiving data: {str(e)} 😢"
 
+def gen_AI_IO_filename(display_query, output):
+    # Get current time in Central Time Zone
+    now_central = datetime.now(pytz.timezone("America/Chicago"))
+    timestamp = now_central.strftime("%Y-%m-%d-%I-%M-%S-%p")
+    
+    # Limit output to first 180 characters and clean up any invalid filename characters
+    output_snippet = re.sub(r'[^A-Za-z0-9]+', '_', output[:180])
+    filename = f"{timestamp} - {display_query} - {output_snippet}.md"
+    return filename
         
 # 🔍 Search glossary - Finding needles in digital haystacks
 def search_glossary(query):
     st.markdown(f"### 🔍 SearchGlossary for: {query}")
-    model_options = ['mistralai/Mixtral-8x7B-Instruct-v0.1', 'mistralai/Mistral-7B-Instruct-v0.2', 'google/gemma-7b-it', 'None']
+    model_options = ['mistralai/Mixtral-8x7B-Instruct-v0.1', 'mistralai/Mistral-7B-Instruct-v0.2']
     model_choice = st.selectbox('🧠 Select LLM Model', options=model_options, index=1, key=f"model_choice_{id(query)}")
     database_options = ['Semantic Search', 'Arxiv Search - Latest - (EXPERIMENTAL)']
     database_choice = st.selectbox('📚 Select Database', options=database_options, index=0, key=f"database_choice_{id(query)}")
@@ -368,33 +377,32 @@ def search_glossary(query):
     #st.code(response2[1], language="python", line_numbers=True, wrap_lines=True)
 
 
-    # Persist AI Results to Markdown Files
+    # ✅ Persist AI Results to Markdown Files
     try:
-        filename = generate_filename(result, "md")
+        filename = gen_AI_IO_filename(display_query, result)
         create_file(filename, query, result)
-        #st.session_state.chat_history.append({"assistant": query, "ArXiV": result})
+        st.markdown(f"✅ File saved as: `{filename}`")
     except:
         st.markdown('1 error')
     try:
-        filename = generate_filename(result2, "md")
+        filename = gen_AI_IO_filename(display_query, result2)
         create_file(filename, query, result2)
-        #st.session_state.chat_history.append({"assistant": query, "ArXiV": result2})
+        st.markdown(f"✅ File saved as: `{filename}`")
     except:
-        st.markdown('2 error')        
+        st.markdown('2 error')
     try:
-        filename = generate_filename(response2[0], "md")
+        filename = gen_AI_IO_filename(display_query, response2[0])
         create_file(filename, query, response2[0])
-        #st.session_state.chat_history.append({"assistant": query, "ArXiV": response2[0]})
+        st.markdown(f"✅ File saved as: `{filename}`")
     except:
         st.markdown('3 error')            
     try:
-        filename = generate_filename(response2[1], "md")
+        filename = gen_AI_IO_filename(display_query, response2[1])
         create_file(filename, query, response2[1])
-        #st.session_state.chat_history.append({"assistant": query, "ArXiV": response2[1]})
+        st.markdown(f"✅ File saved as: `{filename}`")
     except:
         st.markdown('4 error')           
 
-    
     return result, result2, response2