import os
os.environ["STREAMLIT_BROWSER_GATHER_USAGE_STATS"] = "false"
os.environ["STREAMLIT_TELEMETRY_EMAIL"] = ""
os.environ["STREAMLIT_HEADLESS"] = "true"


import streamlit as st
from inference import generate_response

st.set_page_config(page_title="DeepSeek-Coder LoRA", page_icon="🤖")
st.title("🤖 DeepSeek-Coder LoRA Playground")
st.write("Fine-tuned on **NL2SH-ALFA** + user data using QLoRA")

prompt = st.text_area("💬 Enter natural-language command:", height=150)
if st.button("Generate Bash Command"):
    with st.spinner("Thinking..."):
        out = generate_response(prompt)
        st.success(out)
        st.code(out, language="bash")