Update app.py

app.py

@@ -3,14 +3,28 @@ import streamlit as st
 import sys
 import io
 
-def execute_code(code):
+def execute_code(code, user_input):
     """Execute the given code and return the output or error message."""
     old_stdout = sys.stdout  # Backup original stdout
     redirected_output = io.StringIO()  # Create a new string buffer
     sys.stdout = redirected_output  # Redirect stdout to buffer
 
+    # Convert user input into list format
+    input_values = user_input.strip().split("\n")
+
+    # Replace input() calls with user-provided input values
+    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})  # Run the user code with mocked 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 +35,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 provide input values if required.")
 
 code_input = st.text_area("Enter your Python code:", height=200)
+user_input = st.text_area("Enter input values (one per line):", height=100)
 
 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
         st.subheader("Output:")
         st.code(output, language="plaintext")
     else:
@@ -39,19 +54,6 @@ if st.button("Run Code"):
 
 
 
-
-
-
-
-
-
-
-
-
-
-
-
-
 # V1 without gemini api
 
 # import streamlit as st