Update backup.py

backup.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
@@ -106,40 +108,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: 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)
+    # Step 4: Initialize the combined content
+    combined_content = ""
 
-        # 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()
+    # Add Prompt with markdown title and emoji
+    combined_content += "# Prompt 📝\n" + prompt + "\n\n"
 
-            with open(f"{base_filename}-Code.py", 'w') as file:
-                file.write(python_code)
+    # 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)
+            
+            # Add Code Results title with markdown and emoji
+            combined_content += "# Code Results 🚀\n"
 
-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}")
+            # 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 += f"```\n{code_output}\n```\n\n"
+                realtimeEvalResponse = "# Code Results 🚀\n" + "```" + code_output + "```\n\n"
+                st.write(realtimeEvalResponse)
+                
+            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 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)]
 
@@ -225,7 +243,7 @@ def readitaloud(result):
     </html>
     '''
 
-    components.html(documentHTML5, width=1280, height=1024)
+    components.html(documentHTML5, width=800, height=300)
     #return result
 
 def chat_with_model(prompt, document_section, model_choice='gpt-3.5-turbo'):