Update backupapp.py

backupapp.py

@@ -32,14 +32,20 @@ from templates import css, bot_template, user_template
 
 # page config and sidebar declares up front allow all other functions to see global class variables
 st.set_page_config(page_title="GPT Streamlit Document Reasoner", layout="wide")
-should_save = st.sidebar.checkbox("💾 Save")
+should_save = st.sidebar.checkbox("💾 Save", value=True)
 
-def generate_filename(prompt, file_type):
+def generate_filename_old(prompt, file_type):
     central = pytz.timezone('US/Central')
     safe_date_time = datetime.now(central).strftime("%m%d_%H%M")  # Date and time DD-HHMM
     safe_prompt = "".join(x for x in prompt if x.isalnum())[:90]  # Limit file name size and trim whitespace
     return f"{safe_date_time}_{safe_prompt}.{file_type}"  # Return a safe file name
 
+def generate_filename(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", "_")
+    safe_prompt = "".join(x for x in replaced_prompt if x.isalnum() or x == "_")[:90]
+    return f"{safe_date_time}_{safe_prompt}.{file_type}"
 
 def transcribe_audio(openai_key, file_path, model):
     OPENAI_API_URL = "https://api.openai.com/v1/audio/transcriptions"
@@ -77,6 +83,35 @@ def save_and_play_audio(audio_recorder):
     return None
 
 def create_file(filename, prompt, response, should_save=True):
+    if not should_save:
+        return
+
+    # Step 2: Extract base filename without extension
+    base_filename, ext = os.path.splitext(filename)
+
+    # Step 3: Check if the response contains Python code
+    has_python_code = bool(re.search(r"```python([\s\S]*?)```", response))
+
+    # Step 4: Write files based on type
+    if ext in ['.txt', '.htm', '.md']:
+        # Create Prompt file
+        with open(f"{base_filename}-Prompt.txt", 'w') as file:
+            file.write(prompt)
+        
+        # Create Response file
+        with open(f"{base_filename}-Response.md", 'w') as file:
+            file.write(response)
+
+        # Create Code file if Python code is present
+        if has_python_code:
+            # Extract Python code from the response
+            python_code = re.findall(r"```python([\s\S]*?)```", response)[0].strip()
+
+            with open(f"{base_filename}-Code.py", 'w') as file:
+                file.write(python_code)
+
+
+def create_file_old(filename, prompt, response, should_save=True):
     if not should_save:
         return
     if filename.endswith(".txt"):
@@ -484,4 +519,3 @@ with st.sidebar:
             filename = generate_filename(raw, 'txt')
             create_file(filename, raw, '', should_save)
             #create_file(filename, raw, '')
-