Update app.py

app.py

@@ -1,10 +1,13 @@
-#
+
 import streamlit as st
 import subprocess
 import sys
 import io
 import os
 
+# Ensure Java and g++ are installed (for cloud environments)
+os.system("apt update && apt install -y default-jdk g++")
+
 # Function to execute Python code
 def execute_python(code, user_input):
     old_stdout = sys.stdout
@@ -42,7 +45,7 @@ def execute_java(code, user_input):
     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)
+    run_process = subprocess.run(["java", "Main"], cwd=os.getcwd(), 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}"
 
 # Function to execute C++ code
@@ -54,7 +57,7 @@ def execute_cpp(code, user_input):
     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)
+    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
@@ -82,7 +85,6 @@ if st.button("Run Code"):
     else:
         st.warning("⚠️ Please enter some code before running.")
 
-
 # V1 without gemini api
 
 # import streamlit as st