Update app.py

app.py

@@ -1,100 +1,39 @@
 
-
 import streamlit as st
-import subprocess
 import sys
 import io
-import os
-import shutil  # For checking dependencies
-import requests
-import google.generativeai as genai  # Gemini API
-
-# Ensure Java and g++ are installed (for cloud environments)
-if not shutil.which("javac"):
-    os.system("apt-get update && apt-get install -y default-jdk g++")
-
-# Function to execute Python code
-def execute_python(code, user_input):
-    old_stdout = sys.stdout
-    redirected_output = io.StringIO()
-    sys.stdout = redirected_output
-
-    input_values = user_input.strip().split("\n")
-
-    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.")
+
+def execute_code(code):
+    """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
 
     try:
-        exec(code, {"input": mock_input})
-        output = redirected_output.getvalue()
+        exec(code, {})  # Execute the user's code safely
+        output = redirected_output.getvalue()  # Get the output from buffer
     except Exception as e:
-        output = f"Error: {str(e)}"
+        output = f"Error: {str(e)}"  # Capture and display any errors
     finally:
-        sys.stdout = old_stdout
-
-    return output.strip()
-
-# Function to execute Java code
-def execute_java(code, user_input):
-    if not shutil.which("javac"):
-        return "Error: Java JDK is not installed. Please install it using 'apt-get install default-jdk'."
-
-    with open("Main.java", "w") as file:
-        file.write(code)
-
-    compile_process = subprocess.run(["javac", "Main.java"], capture_output=True, text=True)
-    if compile_process.returncode != 0:
-        return f"Compilation Error:\n{compile_process.stderr}"
-
-    run_process = subprocess.run(["java", "Main"], input=user_input.encode(), capture_output=True, text=True)
-    return run_process.stdout if run_process.returncode == 0 else f"Runtime Error:\n{run_process.stderr}"
+        sys.stdout = old_stdout  # Restore original stdout
 
-# Function to execute C++ code
-def execute_cpp(code, user_input):
-    if not shutil.which("g++"):
-        return "Error: C++ Compiler is not installed. Please install it using 'apt-get install g++'."
+    return output.strip()  # Return cleaned output
 
-    with open("main.cpp", "w") as file:
-        file.write(code)
+# Streamlit UI
+st.title("💻 Code Runner")
+st.write("Write your code and get the correct output!")
 
-    compile_process = subprocess.run(["g++", "main.cpp", "-o", "main"], capture_output=True, text=True)
-    if compile_process.returncode != 0:
-        return f"Compilation Error:\n{compile_process.stderr}"
-
-    run_process = subprocess.run(["./main"], input=user_input.encode(), capture_output=True, text=True)
-    return run_process.stdout if run_process.returncode == 0 else f"Runtime Error:\n{run_process.stderr}"
-
-# Streamlit UI for Code Execution
-st.title("💻 Multi-Language Code Runner")
-st.write("Write your Python, Java, or C++ code and get the correct output!")
-
-languages = ["Python", "Java", "C++"]
-selected_lang = st.selectbox("Select Language:", languages)
-
-code_input = st.text_area("Enter your code:", height=200)
-user_input = st.text_area("Enter input values (one per line):", height=100)
+code_input = st.text_area("Enter your Python code:", height=200)
 
 if st.button("Run Code"):
     if code_input.strip():
         with st.spinner("Executing..."):
-            if selected_lang == "Python":
-                output = execute_python(code_input, user_input)
-            elif selected_lang == "Java":
-                output = execute_java(code_input, user_input)
-            elif selected_lang == "C++":
-                output = execute_cpp(code_input, user_input)
-
+            output = execute_code(code_input)  # Execute user code
         st.subheader("Output:")
         st.code(output, language="plaintext")
     else:
-        st.warning("⚠️ Please enter some code before running.")
+        st.warning("⚠️ Please enter some Python code before running.")
+