File size: 3,212 Bytes
f5e6b7b | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 | import streamlit as st
from utils import get_response, apply_tone, apply_language, apply_format
from config import MODEL_LIST
import os
from dotenv import load_dotenv
# 🔐 Load Hugging Face token dari .env atau sidebar
load_dotenv()
hf_token = st.sidebar.text_input("🔑 Hugging Face Token", type="password")
if hf_token:
os.environ["HF_TOKEN"] = hf_token
# 💅 Custom background and styling
st.markdown("""
<style>
.stApp {
background-color: #fff8dc; /* warm yellow */
}
.user-bubble {
background-color: #fef3c7; /* soft yellow */
padding: 10px;
border-radius: 10px;
margin-bottom: 5px;
}
.bot-bubble {
background-color: #e0f2fe; /* soft blue */
padding: 10px;
border-radius: 10px;
margin-bottom: 10px;
}
.element-container pre {
background-color: #1e1e1e !important; /* dark code block */
color: #dcdcdc !important;
border-radius: 8px;
padding: 12px;
font-size: 14px;
}
</style>
""", unsafe_allow_html=True)
# 🧠 Page config
st.set_page_config(page_title="Mini Chatbot", layout="wide")
st.title("💬 Chat with Your LLM")
# 🔧 Sidebar settings
st.sidebar.title("🔧 Settings")
# ❌ Kalau mau aktifin OpenAI lagi, uncomment ini:
# openai_api_key = st.sidebar.text_input("🔑 OpenAI API Key", type="password")
# if openai_api_key:
# os.environ["OPENAI_API_KEY"] = openai_api_key
# use_openai = st.sidebar.checkbox("Use OpenAI backend", value=True)
model_choice = st.sidebar.selectbox("Choose Model", MODEL_LIST)
temperature = st.sidebar.slider("Temperature", 0.0, 1.0, 0.7)
max_tokens = st.sidebar.slider("Max Tokens", 100, 1000, 500)
tone = st.sidebar.selectbox("🎭 Pilih Gaya Bahasa (Tone)", ["Formal", "Casual", "Humorous", "Professional", "Sarcastic"])
language = st.sidebar.selectbox("🌐 Pilih Bahasa Respon", ["Indonesia", "English", "Chinese"])
response_format = st.sidebar.selectbox("🧾 Format Respon", ["Text", "Bullet Points", "Code"])
# 💬 Chat history
if "history" not in st.session_state:
st.session_state.history = []
# 📝 Chat input
user_input = st.chat_input("Type your message here...")
if user_input:
with st.spinner("Thinking..."):
modified_prompt = apply_tone(user_input, tone)
modified_prompt = apply_language(modified_prompt, language)
modified_prompt = apply_format(modified_prompt, response_format)
response = get_response(modified_prompt, "HuggingFace", model_choice, temperature, max_tokens)
decorated_response = f"🤖 {response}"
st.session_state.history.append(("You", user_input))
st.session_state.history.append(("Bot", decorated_response))
# 🖼️ Display chat (user + bot)
for speaker, text in st.session_state.history:
bubble_class = "user-bubble" if speaker == "You" else "bot-bubble"
if speaker == "Bot" and response_format == "Code":
st.code(text.replace("🤖 ", ""), language="python")
else:
st.markdown(f'<div class="{bubble_class}"><strong>{speaker}:</strong> {text}</div>', unsafe_allow_html=True) |