Update app.py

app.py

@@ -1,86 +1,86 @@
-from flask import Flask, render_template, request, jsonify, send_file
+import streamlit as st
 import subprocess
-import tempfile
 import os
+import tempfile
 import shutil
-import requests
 
-app = Flask(__name__)
+# Set up a temporary working directory
+if "temp_dir" not in st.session_state:
+    st.session_state.temp_dir = tempfile.mkdtemp()
 
-# Temporary directory to store downloaded files
-TEMP_DIR = tempfile.mkdtemp()
+temp_dir = st.session_state.temp_dir
 
-@app.route('/')
-def index():
-    return render_template('index.html')
+# 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.route('/execute', methods=['POST'])
-def execute_command():
-    command = request.json['command']
-    
-    if command.startswith('pip install'):
-        return execute_pip_install(command)
-    elif command.startswith('git clone'):
-        return execute_git_clone(command)
-    elif command.startswith('python'):
-        return execute_python_script(command)
-    elif command.startswith('download'):
-        return download_file(command)
-    elif command.startswith('!'):
-        return execute_shell_command(command[1:])
-    else:
-        return jsonify({'output': 'Unknown command. Try pip install, git clone, python, download, or !<shell command>.'})
+# App title
+st.title("Google Colab-like Terminal in Streamlit")
 
-def execute_pip_install(command):
-    try:
-        result = subprocess.run(command, shell=True, check=True, capture_output=True, text=True)
-        return jsonify({'output': result.stdout})
-    except subprocess.CalledProcessError as e:
-        return jsonify({'output': f'Error: {e.stderr}'})
+# Command execution area
+st.subheader("Terminal")
+if "output" not in st.session_state:
+    st.session_state.output = ""
 
-def execute_git_clone(command):
-    try:
-        result = subprocess.run(command, shell=True, check=True, capture_output=True, text=True, cwd=TEMP_DIR)
-        return jsonify({'output': f'Repository cloned successfully to {TEMP_DIR}'})
-    except subprocess.CalledProcessError as e:
-        return jsonify({'output': f'Error: {e.stderr}'})
+# Display terminal output
+st.markdown(f'<div class="terminal">{st.session_state.output}</div>', unsafe_allow_html=True)
 
-def execute_python_script(command):
-    try:
-        with tempfile.NamedTemporaryFile(mode='w', suffix='.py', delete=False, dir=TEMP_DIR) as temp_file:
-            temp_file.write(command[7:])  # Remove 'python ' from the beginning
-            temp_file_path = temp_file.name
-
-        result = subprocess.run(f'python {temp_file_path}', shell=True, check=True, capture_output=True, text=True)
-        os.unlink(temp_file_path)
-        return jsonify({'output': result.stdout})
-    except subprocess.CalledProcessError as e:
-        return jsonify({'output': f'Error: {e.stderr}'})
+# Command input
+command = st.text_input("Enter your command:", placeholder="e.g., pip install numpy")
 
-def download_file(command):
+# Execute the command
+if st.button("Run"):
     try:
-        _, url = command.split(maxsplit=1)
-        response = requests.get(url)
-        if response.status_code == 200:
-            filename = os.path.join(TEMP_DIR, url.split('/')[-1])
-            with open(filename, 'wb') as f:
-                f.write(response.content)
-            return jsonify({'output': f'File downloaded successfully: {filename}'})
-        else:
-            return jsonify({'output': f'Error downloading file: HTTP {response.status_code}'})
+        # 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"
     except Exception as e:
-        return jsonify({'output': f'Error downloading file: {str(e)}'})
+        st.session_state.output += f"$ {command}\nError: {e}\n"
 
-def execute_shell_command(command):
-    try:
-        result = subprocess.run(command, shell=True, check=True, capture_output=True, text=True, cwd=TEMP_DIR)
-        return jsonify({'output': result.stdout})
-    except subprocess.CalledProcessError as e:
-        return jsonify({'output': f'Error: {e.stderr}'})
+# 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.")
 
-@app.teardown_appcontext
-def cleanup(error):
-    shutil.rmtree(TEMP_DIR, ignore_errors=True)
+# Cleanup temporary files when the session ends
+def cleanup():
+    shutil.rmtree(temp_dir, ignore_errors=True)
 
-if __name__ == '__main__':
-    app.run(debug=True)
+st.on_session_end(cleanup)