Update app.py

app.py

@@ -3,11 +3,13 @@ import openai
 import os
 import base64
 import glob
+import io
 import json
 import mistune
 import pytz
 import math
 import requests
+import sys
 import time
 import re
 import textract
@@ -96,6 +98,7 @@ def save_and_play_audio(audio_recorder):
         return filename
     return None
 
+
 def create_file(filename, prompt, response, should_save=True):
     if not should_save:
         return
@@ -106,6 +109,56 @@ def create_file(filename, prompt, response, should_save=True):
     # Step 3: Check if the response contains Python code
     has_python_code = bool(re.search(r"```python([\s\S]*?)```", response))
 
+    # Step 4: Initialize the combined content
+    combined_content = ""
+
+    # Add Prompt with markdown title and emoji
+    combined_content += "# Prompt 📝\n" + prompt + "\n\n"
+
+    # Add Response with markdown title and emoji
+    combined_content += "# Response 💬\n" + response + "\n\n"
+
+    # Check for Python code or other resources and add them with markdown title and emoji
+    resources = re.findall(r"```([\s\S]*?)```", response)
+    for resource in resources:
+        # Check if the resource is Python code
+        if "python" in resource.lower():
+            # Add Code Results title with markdown and emoji
+            combined_content += "# Code Results 🚀\n"
+
+            # Capture standard output
+            original_stdout = sys.stdout
+            sys.stdout = io.StringIO()
+            
+            # Execute Python code and capture the output
+            try:
+                exec(resource)
+                code_output = sys.stdout.getvalue()
+                combined_content += f"```\n{code_output}\n```\n\n"
+            except Exception as e:
+                combined_content += f"```python\nError executing Python code: {e}\n```\n\n"
+            
+            # Restore the original standard output
+            sys.stdout = original_stdout
+        else:
+            # Add Resource title with markdown and emoji for non-Python resources
+            combined_content += "# Resource 🛠️\n" + "```" + resource + "```\n\n"
+
+    # Write the combined content into one file
+    with open(f"{base_filename}-Combined.md", 'w') as file:
+        file.write(combined_content)
+        
+
+def create_file_old(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']: