Update app.py

app.py

@@ -135,10 +135,65 @@ MODEL = "gpt-4o-2024-05-13"
 
 
 
+# 5. Auto name generated output files from time and content
+def generate_filename(prompt, file_type):
+    """
+    Generates a safe filename using the prompt and file type.
+    It allows Unicode characters, including emojis, and replaces unsafe characters with spaces.
+    """
+    # Get current time in the US/Central timezone
+    central = pytz.timezone('US/Central')
+    safe_date_time = datetime.now(central).strftime("%m%d_%H%M")
+
+    # Replace any unsafe characters with spaces, allow emojis and Unicode characters
+    replaced_prompt = re.sub(r'[<>:"/\\|?*\n]', ' ', prompt)
+    
+    # Strip extra spaces from the start and end, and collapse multiple spaces
+    safe_prompt = re.sub(r'\s+', ' ', replaced_prompt).strip()[:240]  # Limit length for filename safety
+
+    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 based on the prompt or content
+    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 the 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):
+    """
+    Extracts 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
 
 
 # 5. Auto name generated output files from time and content
-def generate_filename(prompt, file_type):
+def generate_filename_old2(prompt, file_type):
     central = pytz.timezone('US/Central')
     safe_date_time = datetime.now(central).strftime("%m%d_%H%M")
     replaced_prompt = prompt.replace(" ", "_").replace("\n", "_")
@@ -147,7 +202,7 @@ 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):
+def create_and_save_file_old2(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.