Update app.py

app.py

@@ -1,18 +1,18 @@
 #
 import streamlit as st
+import subprocess
 import sys
 import io
+import os
 
-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
+# Function to execute Python code
+def execute_python(code, user_input):
+    old_stdout = sys.stdout
+    redirected_output = io.StringIO()
+    sys.stdout = redirected_output
 
-    # 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
@@ -24,34 +24,63 @@ def execute_code(code, user_input):
             raise ValueError("Not enough inputs provided.")
 
     try:
-        exec(code, {"input": mock_input})  # Run the user code with mocked input
-        output = redirected_output.getvalue()  # Get the output from buffer
+        exec(code, {"input": mock_input})
+        output = redirected_output.getvalue()
     except Exception as e:
-        output = f"Error: {str(e)}"  # Capture and display any errors
+        output = f"Error: {str(e)}"
     finally:
-        sys.stdout = old_stdout  # Restore original stdout
+        sys.stdout = old_stdout
 
-    return output.strip()  # Return cleaned output
+    return output.strip()
+
+# Function to execute Java code
+def execute_java(code, user_input):
+    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, capture_output=True, text=True)
+    return run_process.stdout if run_process.returncode == 0 else f"Runtime Error:\n{run_process.stderr}"
+
+# Function to execute C++ code
+def execute_cpp(code, user_input):
+    with open("main.cpp", "w") as file:
+        file.write(code)
+
+    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, capture_output=True, text=True)
+    return run_process.stdout if run_process.returncode == 0 else f"Runtime Error:\n{run_process.stderr}"
 
 # Streamlit UI
-st.title("💻 Code Runner")
-st.write("Write your Python code and provide input values if required.")
+st.title("💻 Multi-Language Code Runner")
+st.write("Write your Python, Java, or C++ code and get the correct output!")
 
-code_input = st.text_area("Enter your Python code:", height=200)
+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)
 
 if st.button("Run Code"):
     if code_input.strip():
         with st.spinner("Executing..."):
-            output = execute_code(code_input, user_input)  # Execute user code
+            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)
+
         st.subheader("Output:")
         st.code(output, language="plaintext")
     else:
-        st.warning("⚠️ Please enter some Python code before running.")
-
-
-
-
+        st.warning("⚠️ Please enter some code before running.")
 
 
 # V1 without gemini api