Update app.py

app.py

@@ -99,17 +99,17 @@ def save_and_play_audio(audio_recorder):
     return None
 
 
+# Define a context dictionary to maintain the state between exec calls
+context = {}
+
 def create_file(filename, prompt, response, should_save=True):
     if not should_save:
         return
 
-    # Step 2: Extract base filename without extension
+    # 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
+    # Initialize the combined content
     combined_content = ""
 
     # Add Prompt with markdown title and emoji
@@ -118,31 +118,28 @@ 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
+    # Check for code blocks in the response
     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
+            # Remove the 'python' keyword from 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
+            # Redirect standard output to capture it
             original_stdout = sys.stdout
             sys.stdout = io.StringIO()
             
-            # Execute cleaned Python code and capture the output within exec_context
+            # Execute the cleaned Python code within the context
             try:
-                exec(cleaned_code, exec_context)
+                exec(cleaned_code, 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']()
+                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"
@@ -150,12 +147,13 @@ def create_file(filename, prompt, response, should_save=True):
             # Restore the original standard output
             sys.stdout = original_stdout
         else:
-            # Add Resource title with markdown and emoji for non-Python resources
+            # Add non-Python resources 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)
+    # Save the combined content to a Markdown file
+    if should_save:
+        with open(f"{base_filename}-Combined.md", 'w') as file:
+            file.write(combined_content)
 
 
 def create_file_old2(filename, prompt, response, should_save=True):