Update app.py

app.py

@@ -147,6 +147,42 @@ def generate_filename(prompt, file_type):
     return f"{safe_date_time}_{safe_prompt}.{file_type}"
 
 
+def create_and_save_file(content, file_type="md", prompt=None, is_image=False, should_save=True):
+    """
+    Combines file name generation and file creation into one function.
+    If the file is a markdown file, extracts the title from the content (if available) and uses it for the filename.
+    """
+    if not should_save:
+        return None
+
+    # Step 1: Generate filename
+    filename = generate_filename(prompt if prompt else content, file_type)
+
+    # Step 2: If it's a markdown file, check if it has a title (e.g., # Heading in markdown)
+    if file_type == "md":
+        title_from_content = extract_markdown_title(content)
+        if title_from_content:
+            filename = generate_filename(title_from_content, file_type)
+
+    # Step 3: Save file
+    with open(filename, "w", encoding="utf-8") as f:
+        if is_image:
+            f.write(content)
+        else:
+            f.write(prompt + "\n\n" + content)
+    
+    return filename
+
+
+def extract_markdown_title(content):
+    """
+    Extract the first markdown title (line starting with '#') from the content.
+    """
+    # Use regex to find the first line that starts with '#'
+    title_match = re.search(r'^\s*#\s*(.+)', content, re.MULTILINE)
+    if title_match:
+        return title_match.group(1).strip()
+    return None
 
 def process_text(text_input):
     if text_input:
@@ -168,7 +204,9 @@ def process_text(text_input):
             return_text = completion.choices[0].message.content
             st.write("Assistant: " + return_text)
             filename = generate_filename(text_input, "md")
-            create_file(filename, text_input, return_text, should_save)
+
+            create_and_save_file(return_text, file_type="md", prompt=text_input, is_image=False, should_save=True)
+            #create_file(filename, text_input, return_text, should_save)
             st.session_state.messages.append({"role": "assistant", "content": return_text})
 
         #st.write("Assistant: " + completion.choices[0].message.content)
@@ -816,7 +854,10 @@ def process_text2(MODEL='gpt-4o-2024-05-13', text_input='What is 2+2 and what is
         return_text = completion.choices[0].message.content
         st.write("Assistant: " + return_text)
         filename = generate_filename(text_input, "md")
-        create_file(filename, text_input, return_text, should_save)
+        
+        create_and_save_file(return_text, file_type="md", prompt=text_input, is_image=False, should_save=True) # the new
+        
+        #create_file(filename, text_input, return_text, should_save)
         return return_text
     
 @st.cache_resource
@@ -1599,7 +1640,10 @@ def process_user_input(user_question):
         filename = generate_filename(user_question, 'txt')
         response = message.content
         user_prompt = user_question
-        create_file(filename, user_prompt, response, should_save)       
+
+        create_and_save_file(response, file_type="md", prompt=user_prompt, is_image=False, should_save=True) # the new
+
+        #create_file(filename, user_prompt, response, should_save)       
 
 def divide_prompt(prompt, max_length):
     words = prompt.split()
@@ -1794,13 +1838,6 @@ if example_input:
 for example_input in session_state["search_queries"]:
     st.write(example_input)
 
-#if st.button("Run Prompt", help="Click to run."):
-#    try:
-#        response=StreamLLMChatResponse(example_input)
-#        create_file(filename, example_input, response, should_save)
-#    except:
-#        st.write('model is asleep. Starting now on A10 GPU.  Please wait one minute then retry.  KEDA triggered.')
-        
 openai.api_key = os.getenv('OPENAI_API_KEY')
 if openai.api_key == None: openai.api_key = st.secrets['OPENAI_API_KEY']
 menu = ["txt", "htm", "xlsx", "csv", "md", "py"]