Update app.py

app.py

@@ -1,4 +1,6 @@
 import streamlit as st
+import io
+import sys
 
 # Set up the layout of the app
 st.title("Intro to Python: Interactive Learning")
@@ -9,6 +11,24 @@ def create_explanation(title, content):
     with st.expander(f"Explanation: {title}"):
         st.markdown(content)
 
+# Helper function to execute code and capture output
+def run_code(user_code):
+    # Create StringIO object to capture output
+    output = io.StringIO()
+    # Redirect stdout to capture print statements
+    sys.stdout = output
+    try:
+        # Execute the user's code
+        exec(user_code, {})
+        # Return the captured output
+        result = output.getvalue()
+        return result, None
+    except Exception as e:
+        return None, str(e)
+    finally:
+        # Reset stdout
+        sys.stdout = sys.__stdout__
+
 # 1. Basic Syntax and Variables
 st.subheader("1. Basic Syntax and Variables")
 code1 = '''name = "Alice"
@@ -33,12 +53,12 @@ st.write("### Try it Yourself:")
 user_code1 = st.text_area("Example 1: Paste or modify the code and hit 'Run' to see the output.", value=code1, height=150, key="1")
 
 if st.button("Run Example 1 Code"):
-    try:
-        exec_locals = {}
-        exec(user_code1, {}, exec_locals)
+    result, error = run_code(user_code1)
+    if result:
         st.success("Code ran successfully!")
-    except Exception as e:
-        st.error(f"Error: {str(e)}")
+        st.text(result)
+    elif error:
+        st.error(f"Error: {error}")
 
 # 2. Control Flow: If-Else Statements
 st.subheader("2. Control Flow: If-Else Statements")
@@ -65,12 +85,12 @@ st.write("### Try it Yourself:")
 user_code2 = st.text_area("Example 2: Paste or modify the code and hit 'Run' to see the output.", value=code2, height=150, key="2")
 
 if st.button("Run Example 2 Code"):
-    try:
-        exec_locals = {}
-        exec(user_code2, {}, exec_locals)
+    result, error = run_code(user_code2)
+    if result:
         st.success("Code ran successfully!")
-    except Exception as e:
-        st.error(f"Error: {str(e)}")
+        st.text(result)
+    elif error:
+        st.error(f"Error: {error}")
 
 # 3. Loops: For and While
 st.subheader("3. Loops: For and While")
@@ -98,12 +118,12 @@ st.write("### Try it Yourself:")
 user_code3 = st.text_area("Example 3: Paste or modify the code and hit 'Run' to see the output.", value=code3, height=150, key="3")
 
 if st.button("Run Example 3 Code"):
-    try:
-        exec_locals = {}
-        exec(user_code3, {}, exec_locals)
+    result, error = run_code(user_code3)
+    if result:
         st.success("Code ran successfully!")
-    except Exception as e:
-        st.error(f"Error: {str(e)}")
+        st.text(result)
+    elif error:
+        st.error(f"Error: {error}")
 
 # 4. Functions: Defining and Using Functions
 st.subheader("4. Functions: Defining and Using Functions")
@@ -129,12 +149,12 @@ st.write("### Try it Yourself:")
 user_code4 = st.text_area("Example 4: Paste or modify the code and hit 'Run' to see the output.", value=code4, height=150, key="4")
 
 if st.button("Run Example 4 Code"):
-    try:
-        exec_locals = {}
-        exec(user_code4, {}, exec_locals)
+    result, error = run_code(user_code4)
+    if result:
         st.success("Code ran successfully!")
-    except Exception as e:
-        st.error(f"Error: {str(e)}")
+        st.text(result)
+    elif error:
+        st.error(f"Error: {error}")
 
 # 5. Lists and Dictionaries
 st.subheader("5. Basic Data Structures: Lists and Dictionaries")
@@ -159,12 +179,12 @@ st.write("### Try it Yourself:")
 user_code5 = st.text_area("Example 5: Paste or modify the code and hit 'Run' to see the output.", value=code5, height=150, key="5")
 
 if st.button("Run Example 5 Code"):
-    try:
-        exec_locals = {}
-        exec(user_code5, {}, exec_locals)
+    result, error = run_code(user_code5)
+    if result:
         st.success("Code ran successfully!")
-    except Exception as e:
-        st.error(f"Error: {str(e)}")
+        st.text(result)
+    elif error:
+        st.error(f"Error: {error}")
 
 # Footer
 st.write("This app is an educational tool to help you learn basic Python concepts. Keep exploring and practicing to master the language!")