Update app.py

app.py

@@ -5,6 +5,71 @@ from contextlib import redirect_stdout
 import re
 import mistune
 
+# Default example code
+DEFAULT_CODE = '''import streamlit as st
+import random
+
+# Set page config
+st.set_page_config(page_title="Test App", page_icon="🧪")
+
+# Title and introduction
+st.title("🧪 Test Program")
+st.markdown("Welcome to this simple test application!")
+
+# Create two columns
+col1, col2 = st.columns(2)
+
+with col1:
+    # Number input
+    number = st.number_input("Enter a number:", min_value=1, max_value=100, value=50)
+    
+    # Slider
+    multiplier = st.slider("Select multiplier:", 1, 10, 5)
+    
+    # Calculate and display result
+    if st.button("Calculate"):
+        result = number * multiplier
+        st.success(f"Result: {result}")
+
+with col2:
+    # Color picker
+    color = st.color_picker("Pick a color", "#00f900")
+    
+    # Display a box with the selected color
+    st.markdown(
+        f"""
+        <div style="background-color: {color}; padding: 20px; border-radius: 5px;">
+            Selected color: {color}
+        </div>
+        """,
+        unsafe_allow_html=True
+    )
+    
+    # Random number generator
+    if st.button("Generate Random Number"):
+        random_num = random.randint(1, 100)
+        st.balloons()
+        st.write(f"Random number: {random_num}")
+
+# Add a text input with a toggle
+show_text_input = st.toggle("Show text input")
+if show_text_input:
+    user_text = st.text_input("Enter some text:")
+    if user_text:
+        st.write("You entered:", user_text)
+
+# Add an expander with additional information
+with st.expander("About this app"):
+    st.write("""
+    This is a simple test application demonstrating various Streamlit features:
+    - Number inputs and sliders
+    - Calculations
+    - Color picker
+    - Random number generation
+    - Text input with toggle
+    - Columns layout
+    """)'''
+
 def extract_python_code(markdown_text):
     """Extract Python code blocks from markdown text."""
     pattern = r"```python\s*(.*?)\s*```"
@@ -40,51 +105,68 @@ def main():
     st.markdown("Enter Python code directly or upload a .py/.md file")
     
     # File uploader
-    uploaded_file = st.file_uploader("Upload a Python (.py) or Markdown (.md) file", 
+    uploaded_file = st.file_uploader("📤 Upload a Python (.py) or Markdown (.md) file", 
                                    type=['py', 'md'])
     
+    # Initialize session state for code if not exists
+    if 'code' not in st.session_state:
+        st.session_state.code = DEFAULT_CODE
+    
     # Code input area
     if uploaded_file is None:
-        code = st.text_area("Or enter your Python code here:", 
-                           height=200,
-                           placeholder="Enter your Python code...")
+        code_input = st.text_area("💻 Python Code Editor:", 
+                                value=st.session_state.code,
+                                height=400,
+                                key="code_editor")
     else:
         content = uploaded_file.getvalue().decode()
         if uploaded_file.type == "text/markdown":
             # Extract Python code from markdown
             code_blocks = extract_python_code(content)
             if code_blocks:
-                code = code_blocks[0]  # Use the first Python code block found
+                code_input = code_blocks[0]  # Use the first Python code block found
             else:
-                st.error("No Python code blocks found in the markdown file!")
+                st.error("❌ No Python code blocks found in the markdown file!")
                 return
         else:
             # Assume it's a Python file
-            code = content
+            code_input = content
         
-        st.code(code, language='python')
+        st.code(code_input, language='python')
+    
+    # Button columns
+    col1, col2, col3 = st.columns([1, 1, 4])
     
-    # Execute button
-    if st.button("▶️ Run Code"):
-        if code:
-            st.markdown("### Output:")
-            output, error = execute_code(code)
-            
-            if error:
-                st.error(f"Error:\n{error}")
-            elif output:
-                st.code(output)
+    with col1:
+        # Execute button
+        if st.button("▶️ Run Code"):
+            if code_input:
+                st.markdown("### 📋 Output:")
+                output, error = execute_code(code_input)
+                
+                if error:
+                    st.error(f"❌ Error:\n{error}")
+                elif output:
+                    st.code(output)
+                else:
+                    st.success("✅ Code executed successfully with no output.")
             else:
-                st.info("Code executed successfully with no output.")
-        else:
-            st.warning("Please enter some code to execute!")
+                st.warning("⚠️ Please enter some code to execute!")
+    
+    with col2:
+        # Clear button
+        if st.button("🗑️ Clear Code"):
+            st.session_state.code = ""
+            st.rerun()
     
     # Add some usage information
     with st.expander("ℹ️ How to use"):
         st.markdown("""
         1. Either upload a Python (.py) or Markdown (.md) file containing Python code
         2. Or enter Python code directly in the text area
-        3. Click the 'Run Code' button to execute
+        3. Use the buttons below the editor to:
+           - ▶️ Run Code: Execute the current code
+           - 🗑️ Clear Code: Clear the editor
         
         **Note:** For markdown files, the code should be in Python code blocks like:
         ````