Update app.py

app.py

@@ -83,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"):