Update app.py

app.py

@@ -668,6 +668,21 @@ def search_arxiv(query):
 
     return responseall
 
+# Function to generate a filename based on prompt and time (because names matter 🕒)
+def generate_filename(prompt, file_type):
+    central = pytz.timezone('US/Central')
+    safe_date_time = datetime.now(central).strftime("%m%d_%H%M")
+    safe_prompt = re.sub(r'\W+', '_', prompt)[:90]
+    return f"{safe_date_time}_{safe_prompt}.{file_type}"
+
+# Function to create and save a file (and avoid the black hole of lost data 🕳)
+def create_file(filename, prompt, response, should_save=True):
+    if not should_save:
+        return
+    with open(filename, 'w', encoding='utf-8') as file:
+        file.write(prompt + "\n\n" + response)
+
+    
 
 def perform_ai_lookup(query):
     start_time = time.strftime("%Y-%m-%d %H:%M:%S")
@@ -708,6 +723,8 @@ def perform_ai_lookup(query):
     st.write(f"Start time: {start_time}")
     st.write(f"Finish time: {end_time}")
     st.write(f"Elapsed time: {elapsed_seconds:.2f} seconds")
+
+    
     filename = generate_filename(query, "md")
     create_file(filename, query, results, should_save)
     return results