Update app.py

app.py

@@ -98,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
@@ -117,6 +118,70 @@ 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)
+    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)
+            
+            # Capture standard output
+            original_stdout = sys.stdout
+            sys.stdout = io.StringIO()
+            
+            # Execute cleaned Python code and capture the output
+            try:
+                exec(cleaned_code)
+                code_output = sys.stdout.getvalue()
+                combined_content += "# Code Results 🚀\n" + "```" + code_output + "```\n\n"
+            except Exception as exec_exception:
+                # If execution fails, write the Python code to a file and execute it as a separate process
+                python_filename = f"{base_filename}.py"
+                with open(python_filename, 'w') as python_file:
+                    python_file.write(cleaned_code)
+                
+                # Run the Python file and capture its output
+                try:
+                    code_output = subprocess.check_output(['python', python_filename], stderr=subprocess.STDOUT, text=True)
+                    pythonFileExecResponse = "# Python File Results 🚀\n" + "```" + code_output + "```\n\n"
+                    combined_content += pythonFileExecResponse
+                except subprocess.CalledProcessError as e:
+                    combined_content += f"```python\nError executing Python file: {e.output}\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
+    combined_filename = f"{base_filename}-Combined.md"
+    with open(combined_filename, 'w') as file:
+        file.write(combined_content)
+    
+    return combined_filename, python_filename if 'python_filename' in locals() else None
+
+
+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: 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: