Update app.py

app.py

@@ -1,104 +1,86 @@
 import streamlit as st
 import subprocess
 import os
+import tempfile
 import shutil
-import uuid
-from pathlib import Path
-import atexit
 
-# Generate a unique workspace for the user
-session_id = str(uuid.uuid4())
-workspace_dir = f"temp_workspace_{session_id}"
-Path(workspace_dir).mkdir(parents=True, exist_ok=True)
+# Set up a temporary working directory
+if "temp_dir" not in st.session_state:
+    st.session_state.temp_dir = tempfile.mkdtemp()
 
-# Helper function to execute shell commands
-def run_shell_command(command):
-    try:
-        result = subprocess.check_output(command, shell=True, stderr=subprocess.STDOUT, text=True)
-        return result
-    except subprocess.CalledProcessError as e:
-        return f"Error: {e.output}"
-
-# Cleanup workspace when the script exits
-def cleanup_workspace():
-    if os.path.exists(workspace_dir):
-        shutil.rmtree(workspace_dir)
-atexit.register(cleanup_workspace)
+temp_dir = st.session_state.temp_dir
 
-# Custom CSS for styling
-st.markdown(
-    """
+# 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;
-        border: none;
         padding: 10px 20px;
-        text-align: center;
-        text-decoration: none;
-        font-size: 16px;
-        margin: 4px 2px;
-        cursor: pointer;
+        border: none;
         border-radius: 5px;
-    }
-    .stButton>button:hover {
-        background-color: #45a049;
-    }
-    textarea {
-        font-family: "Courier New", monospace;
-        font-size: 14px;
+        cursor: pointer;
     }
     </style>
-    """,
-    unsafe_allow_html=True,
-)
-
-st.title("Advanced Streamlit IDE")
+""", unsafe_allow_html=True)
 
-# Editor for writing Python code
-st.subheader("Code Editor")
-default_code = """import streamlit as st
+# App title
+st.title("Google Colab-like Terminal in Streamlit")
 
-st.title("Welcome to the Advanced Streamlit IDE!")
-st.write("Start coding your Python project here.")"""
-code = st.text_area("Write your Python code below:", value=default_code, height=300)
+# Command execution area
+st.subheader("Terminal")
+if "output" not in st.session_state:
+    st.session_state.output = ""
 
-# Option to install packages
-st.subheader("Install Python Packages")
-packages = st.text_input("Enter package names separated by space (e.g., numpy pandas):")
+# Display terminal output
+st.markdown(f'<div class="terminal">{st.session_state.output}</div>', unsafe_allow_html=True)
 
-if st.button("Install Packages"):
-    with st.spinner("Installing packages..."):
-        install_command = f"pip install --target={workspace_dir} {packages}"
-        install_output = run_shell_command(install_command)
-    st.text(install_output)
+# Command input
+command = st.text_input("Enter your command:", placeholder="e.g., pip install numpy")
 
-# Run the code in a temporary workspace
-if st.button("Run Code"):
-    temp_file_path = os.path.join(workspace_dir, "temp_app.py")
-    with open(temp_file_path, "w") as temp_file:
-        temp_file.write(code)
-    
-    with st.spinner("Running your code..."):
-        run_command = f"PYTHONPATH={workspace_dir} python {temp_file_path}"
-        run_output = run_shell_command(run_command)
-    st.text(run_output)
+# 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"
+    except Exception as e:
+        st.session_state.output += f"$ {command}\nError: {e}\n"
 
-# Export code
-if st.button("Export Code"):
-    temp_file_path = os.path.join(workspace_dir, "temp_app.py")
-    if os.path.exists(temp_file_path):
-        with open(temp_file_path, "r") as temp_file:
-            exported_code = temp_file.read()
-        st.download_button("Download Code", data=exported_code, file_name="exported_code.py", mime="text/plain")
+# 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.error("No code to export!")
+        st.write("No files found.")
+
+# Cleanup temporary files when the session ends
+def cleanup():
+    shutil.rmtree(temp_dir, ignore_errors=True)
 
-st.markdown("---")
-st.info(
-    """
-    - Write and execute Python code in a temporary workspace.
-    - Install and use packages dynamically within your session.
-    - All data is deleted once you leave the site.
-    """
-)
+st.on_session_end(cleanup)