Update app.py

app.py

@@ -122,6 +122,9 @@ def create_file(filename, prompt, response, should_save=True):
     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"
 
@@ -129,9 +132,9 @@ def create_file(filename, prompt, response, should_save=True):
             original_stdout = sys.stdout
             sys.stdout = io.StringIO()
             
-            # Execute Python code and capture the output
+            # Execute cleaned Python code and capture the output
             try:
-                exec(resource)
+                exec(cleaned_code)
                 code_output = sys.stdout.getvalue()
                 combined_content += f"```\n{code_output}\n```\n\n"
             except Exception as e:
@@ -145,58 +148,13 @@ def create_file(filename, prompt, response, should_save=True):
 
     # 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))
+        file.write(combined_content)        
 
-    
-    # Step 4: Write files based on type
-    if ext in ['.txt', '.htm', '.md']:
-        # 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:
-            # Add Resource title with markdown and emoji
-            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
-    if filename.endswith(".txt"):
-        with open(filename, 'w') as file:
-            file.write(f"{prompt}\n{response}")
-    elif filename.endswith(".htm"):
-        with open(filename, 'w') as file:
-            file.write(f"{prompt}   {response}")
-    elif filename.endswith(".md"):
-        with open(filename, 'w') as file:
-            file.write(f"{prompt}\n\n{response}")
-            
 def truncate_document(document, length):
     return document[:length]
+
 def divide_document(document, max_length):
     return [document[i:i+max_length] for i in range(0, len(document), max_length)]