Update app.py

app.py

@@ -99,7 +99,6 @@ def save_and_play_audio(audio_recorder):
     return None
 
 
-
 def create_file(filename, prompt, response, should_save=True):
     if not should_save:
         return
@@ -119,6 +118,65 @@ def create_file(filename, prompt, response, should_save=True):
     # 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)
+    # Create a dictionary to store exec() context
+    exec_context = {}
+    for resource in resources:
+        # Check if the resource contains Python code
+        if "python" in resource.lower():
+            # Remove the word 'python' from the beginning of the code block
+            cleaned_code = re.sub(r'^\s*python', '', resource, flags=re.IGNORECASE | re.MULTILINE)
+            
+            # 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 cleaned Python code and capture the output within exec_context
+            try:
+                exec(cleaned_code, exec_context)
+                code_output = sys.stdout.getvalue()
+                combined_content += f"```\n{code_output}\n```\n\n"
+                # Use the exec_context here to access any defined functions or variables
+                # For example, if 'cleaned_code' defined a function 'foo', you can call it:
+                # exec_context['foo']()
+                
+            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_old2(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: 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: