Update app.py

app.py

@@ -1,16 +1,27 @@
-
 import streamlit as st
 import sys
 import io
 
-def execute_code(code):
-    """Execute the given code and return the output or error message."""
+def execute_code(code, user_input=""):
+    """Execute the given code with simulated input and return the output."""
     old_stdout = sys.stdout  # Backup original stdout
     redirected_output = io.StringIO()  # Create a new string buffer
     sys.stdout = redirected_output  # Redirect stdout to buffer
 
+    input_values = user_input.strip().split("\n")  # Split user inputs by line
+    input_counter = 0  
+
+    def mock_input(prompt=""):
+        nonlocal input_counter
+        if input_counter < len(input_values):
+            value = input_values[input_counter]
+            input_counter += 1
+            return value
+        else:
+            raise ValueError("Not enough inputs provided.")
+
     try:
-        exec(code, {})  # Execute the user's code safely
+        exec(code, {"input": mock_input})  # Execute the user's code with mock input
         output = redirected_output.getvalue()  # Get the output from buffer
     except Exception as e:
         output = f"Error: {str(e)}"  # Capture and display any errors
@@ -21,14 +32,15 @@ def execute_code(code):
 
 # Streamlit UI
 st.title("💻 Code Runner")
-st.write("Write your code and get the correct output!")
+st.write("Write your Python code and get the correct output!")
 
 code_input = st.text_area("Enter your Python code:", height=200)
+user_input = st.text_area("Enter input values (one per line):", height=100)  # Added input field
 
 if st.button("Run Code"):
     if code_input.strip():
         with st.spinner("Executing..."):
-            output = execute_code(code_input)  # Execute user code
+            output = execute_code(code_input, user_input)  # Execute user code with mock input
         st.subheader("Output:")
         st.code(output, language="plaintext")
     else:
@@ -43,13 +55,6 @@ if st.button("Run Code"):
 
 
 
-
-
-
-
-
-
-
         
 # V1 without gemini api