Update pages/3_WithExercises.py

pages/3_WithExercises.py

@@ -22,7 +22,7 @@ def execute_code(code):
     
     return output, local_vars
 
-
+# Dictionary with exercise details
 exercises = {
     "Exercise 1: Create and Manipulate Tensors": {
         "description": "Tensors are the core data structure in PyTorch, similar to arrays in NumPy but with additional capabilities for GPU acceleration. This exercise introduces how to create tensors from various data sources such as lists and NumPy arrays. It also covers basic tensor operations like addition, subtraction, and element-wise multiplication, which are fundamental for manipulating data in PyTorch.",
@@ -364,7 +364,7 @@ code_input = st.text_area("Enter your PyTorch code here", height=300, value=exer
 # Button to execute the code
 if st.button("Run Code"):
     # Prepend the import statement
-    code_to_run = "import torch" + code_input
+    code_to_run = "import torch\n" + code_input
     
     # Execute the code and capture the output
     output, variables = execute_code(code_to_run)
@@ -374,8 +374,7 @@ if st.button("Run Code"):
     st.text(output)
     
     # Display returned variables
-    #if variables:
-        #st.subheader('Variables')
-        #for key, value in variables.items():
-            #st.text(f"{key}: {value}")
-            
+    if variables:
+        st.subheader('Variables')
+        for key, value in variables.items():
+            st.text(f"{key}: {value}")