Update app.py

app.py

@@ -15,14 +15,10 @@ import re
 import textract
 import zipfile  
 import random
-
 import httpx # add 11/13/23
 import asyncio
 from openai import OpenAI
 #from openai import AsyncOpenAI
-
-
-
 from datetime import datetime
 from xml.etree import ElementTree as ET
 from bs4 import BeautifulSoup
@@ -40,8 +36,11 @@ from templates import css, bot_template, user_template
 from io import BytesIO
 from contextlib import redirect_stdout
 
-# page config and sidebar declares up front allow all other functions to see global class variables
+
+# set page config once
 st.set_page_config(page_title="Python AI Pair Programmer", layout="wide")
+
+# UI for sidebar controls
 should_save = st.sidebar.checkbox("💾 Save", value=True)
 col1, col2, col3, col4 = st.columns(4)
 with col1:
@@ -51,12 +50,9 @@ with col1:
         choice = st.sidebar.selectbox("Output File Type:", menu)
         model_choice = st.sidebar.radio("Select Model:", ('gpt-3.5-turbo', 'gpt-3.5-turbo-0301'))
 
-        
 # 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
@@ -94,7 +90,7 @@ def create_file(filename, prompt, response, should_save=True):
                 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)
+                st.code(realtimeEvalResponse)
                 
             except Exception as e:
                 combined_content += f"```python\nError executing Python code: {e}\n```\n\n"
@@ -109,7 +105,7 @@ def create_file(filename, prompt, response, should_save=True):
     if should_save:
         with open(f"{base_filename}.md", 'w') as file:
             file.write(combined_content)
-            st.write(combined_content)
+            st.code(combined_content)
 
     # Create a Base64 encoded link for the file
     with open(f"{base_filename}.md", 'rb') as file: