MistriDevLab / streamlit_run.py
acecalisto3's picture
Rename app.py to streamlit_run.py
959d363 verified
import streamlit as st
import gradio as gr
from transformers import pipeline, AutoModelForSeq2SeqLM, AutoTokenizer
import subprocess
import os
# Initialize Hugging Face pipelines
text_generator = pipeline("text-generation", model="gpt2")
code_generator = pipeline("text2text-generation", model="t5-base")
# Streamlit App
st.title("AI Dev Tool Kit")
# Sidebar for Navigation
st.sidebar.title("Navigation")
app_mode = st.sidebar.selectbox("Choose the app mode", ["Explorer", "In-Chat Terminal", "Tool Box"])
if app_mode == "Explorer":
st.header("Explorer")
st.write("Explore files and projects here.")
# Implement your explorer functionality here
elif app_mode == "In-Chat Terminal":
st.header("In-Chat Terminal")
def run_terminal_command(command):
try:
result = subprocess.run(command, shell=True, capture_output=True, text=True)
return result.stdout if result.returncode == 0 else result.stderr
except Exception as e:
return str(e)
def terminal_interface(command):
response = run_terminal_command(command)
return response
def nlp_code_interpreter(text):
response = code_generator(text, max_length=150)
code = response[0]['generated_text']
return code, run_terminal_command(code)
with gr.Blocks() as iface:
terminal_input = gr.Textbox(label="Enter Command or Code")
terminal_output = gr.Textbox(label="Terminal Output", lines=10)
terminal_button = gr.Button("Run")
terminal_button.click(
nlp_code_interpreter,
inputs=terminal_input,
outputs=[terminal_output, terminal_output]
)
iface.launch()
st.write("Use the terminal to execute commands or interpret natural language into code.")
elif app_mode == "Tool Box":
st.header("Tool Box")
st.write("Access various AI development tools here.")
# Implement your tool box functionality here
# Deploy to Hugging Face Spaces
def deploy_to_huggingface(app_name):
code = f"""
import gradio as gr
def run_terminal_command(command):
try:
result = subprocess.run(command, shell=True, capture_output=True, text=True)
return result.stdout if result.returncode == 0 else result.stderr
except Exception as e:
return str(e)
def nlp_code_interpreter(text):
response = code_generator(text, max_length=150)
code = response[0]['generated_text']
return code, run_terminal_command(code)
with gr.Blocks() as iface:
terminal_input = gr.Textbox(label="Enter Command or Code")
terminal_output = gr.Textbox(label="Terminal Output", lines=10)
terminal_button = gr.Button("Run")
terminal_button.click(
nlp_code_interpreter,
inputs=terminal_input,
outputs=[terminal_output, terminal_output]
)
iface.launch()
"""
with open("app.py", "w") as f:
f.write(code)
try:
subprocess.run(["huggingface-cli", "repo", "create", "--type", "space", "--space_sdk", "gradio", app_name], check=True)
subprocess.run(["git", "init"], cwd=f"./{app_name}", check=True)
subprocess.run(["git", "add", "."], cwd=f"./{app_name}", check=True)
subprocess.run(['git', 'commit', '-m', '"Initial commit"'], cwd=f'./{app_name}', check=True)
subprocess.run(["git", "push", "https://huggingface.co/spaces/" + app_name, "main"], cwd=f'./{app_name}', check=True)
return f"Successfully deployed to Hugging Face Spaces: https://huggingface.co/spaces/{app_name}"
except Exception as e:
return f"Error deploying to Hugging Face Spaces: {e}"
# Example usage
if st.button("Deploy to Hugging Face"):
app_name = "ai-dev-toolkit"
deploy_status = deploy_to_huggingface(app_name)
st.write(deploy_status)