Update app.py

app.py

@@ -1,86 +1,40 @@
-import streamlit as st
-import subprocess
-import os
-import tempfile
-import shutil
+from flask import Flask, request, jsonify
+import sys
+import io
+import contextlib
 
-# Set up a temporary working directory
-if "temp_dir" not in st.session_state:
-    st.session_state.temp_dir = tempfile.mkdtemp()
+app = Flask(__name__)
 
-temp_dir = st.session_state.temp_dir
+# Function to execute Python code safely
+def execute_python_code(code):
+    # Capturing the output of Python execution
+    output = io.StringIO()
+    sys.stdout = output
+    sys.stderr = output
 
-# Custom CSS for Google Colab-like design
-st.markdown("""
-    <style>
-    .terminal {
-        background-color: #1e1e1e;
-        color: #00ff00;
-        font-family: monospace;
-        padding: 10px;
-        border-radius: 5px;
-        overflow-y: auto;
-        max-height: 300px;
-    }
-    .terminal-input {
-        background-color: #1e1e1e;
-        color: #00ff00;
-        font-family: monospace;
-        border: none;
-        outline: none;
-        padding: 10px;
-        width: 100%;
-        border-radius: 5px;
-    }
-    .stButton>button {
-        background-color: #4CAF50;
-        color: white;
-        padding: 10px 20px;
-        border: none;
-        border-radius: 5px;
-        cursor: pointer;
-    }
-    </style>
-""", unsafe_allow_html=True)
-
-# App title
-st.title("Google Colab-like Terminal in Streamlit")
-
-# Command execution area
-st.subheader("Terminal")
-if "output" not in st.session_state:
-    st.session_state.output = ""
-
-# Display terminal output
-st.markdown(f'<div class="terminal">{st.session_state.output}</div>', unsafe_allow_html=True)
-
-# Command input
-command = st.text_input("Enter your command:", placeholder="e.g., pip install numpy")
-
-# Execute the command
-if st.button("Run"):
     try:
-        # Execute the command and capture output
-        result = subprocess.run(command, shell=True, cwd=temp_dir, capture_output=True, text=True)
-        output = result.stdout if result.returncode == 0 else result.stderr
-        st.session_state.output += f"$ {command}\n{output}\n"
+        # Executing the code
+        exec(code)
     except Exception as e:
-        st.session_state.output += f"$ {command}\nError: {e}\n"
-
-# File management
-st.subheader("Files")
-if st.button("Show Files"):
-    files = os.listdir(temp_dir)
-    if files:
-        st.write("Files in the working directory:")
-        for file in files:
-            file_path = os.path.join(temp_dir, file)
-            st.download_button(label=f"Download {file}", data=open(file_path, "rb"), file_name=file)
-    else:
-        st.write("No files found.")
-
-# Cleanup temporary files when the session ends
-def cleanup():
-    shutil.rmtree(temp_dir, ignore_errors=True)
-
-st.on_session_end(cleanup)
+        output.write(f"Error: {e}")
+
+    # Return the captured output
+    return output.getvalue()
+
+@app.route('/')
+def index():
+    return app.send_static_file('index.html')
+
+@app.route('/execute', methods=['POST'])
+def execute():
+    data = request.get_json()
+    code = data.get('code', '')
+    
+    # Execute the Python code and capture the result
+    result = execute_python_code(code)
+    
+    # Send the result back to the frontend
+    return jsonify({'result': result})
+
+if __name__ == '__main__':
+    app.run(debug=True)