fix app

app.py

@@ -1,7 +1,20 @@
 import subprocess
+import torch
+import streamlit as st
+import streamlit_ace
 
+try:
+    for i in range(torch.cuda.device_count()):
+        st.text(torch.cuda.get_device_properties(i).name)
+except:
+    st.text("Could not get device properties")
 input_user = ""
-while input_user != "Q":
-    if input_user:
-        (subprocess.run(input_user.rstrip(), capture_output=True).stdout)
-    input_user = input()
+with st.form("input code"):
+    code: str = streamlit_ace.st_ace(auto_update=True)
+    if st.form_submit_button("run code"):
+        prossess = subprocess.run(
+            code.rstrip().split(), shell=True, capture_output=True
+        )
+        st.text(prossess.stdout.decode("utf-8"))
+        if prossess.returncode == 0:
+            st.error(prossess.stderr.decode("utf-8"))

requirements.txt

@@ -4,4 +4,4 @@ sentencepiece==0.1.*
 torch==2.*
 transformers==4.*
 uvicorn[standard]==0.17.*
-xformers==0.0.19
+streamlit-ace

test.py

@@ -0,0 +1,3 @@
+import subprocess
+
+print(subprocess.run(["dir"], shell=True, capture_output=True).stdout)