Update app.py

app.py

@@ -30,9 +30,23 @@ def chat(user_input, history):
     history.append((user_input, response))
     return history, history
 
-def execute_code(code):
+def execute_code(code, user_inputs):
+    # Split user inputs by newline
+    inputs = user_inputs.strip().split('\n')
+    input_iter = iter(inputs)
+    
+    # Custom input function to replace built-in input()
+    def custom_input(prompt=''):
+        try:
+            return next(input_iter)
+        except StopIteration:
+            raise Exception("Not enough inputs provided.")
+    
+    # Redirect stdout
     old_stdout = sys.stdout
     redirected_output = sys.stdout = io.StringIO()
+    old_input = __builtins__.input
+    __builtins__.input = custom_input
     try:
         exec(code, {})
         output = redirected_output.getvalue()
@@ -40,6 +54,7 @@ def execute_code(code):
         output = f"Error: {e}\n{traceback.format_exc()}"
     finally:
         sys.stdout = old_stdout
+        __builtins__.input = old_input
     return output
 
 with gr.Blocks() as demo:
@@ -50,10 +65,12 @@ with gr.Blocks() as demo:
         msg.submit(chat, inputs=[msg, chatbot], outputs=[chatbot, chatbot])
     with gr.Tab("Interpreter"):
         gr.Markdown("### 🖥️ Test Your Code")
-        code_input = gr.Code(language="python")
+        code_input = gr.Code(language="python", lines=20)
+        gr.Markdown("#### 📝 Provide Inputs (Each input on a new line):")
+        user_inputs = gr.Textbox(lines=5, placeholder="Enter inputs for your code here...")
         run_button = gr.Button("Run Code")
-        code_output = gr.Textbox(label="Output")
-        run_button.click(execute_code, inputs=code_input, outputs=code_output)
+        code_output = gr.Textbox(label="Output", lines=15)
+        run_button.click(execute_code, inputs=[code_input, user_inputs], outputs=code_output)
     with gr.Tab("Logs"):
         gr.Markdown("### 📜 Logs")
         log_output = gr.Textbox(label="Logs", lines=10, interactive=False)