import streamlit as st
import torch
from transformers import AutoTokenizer, AutoModelForCausalLM
from peft import PeftModel, PeftConfig
from huggingface_hub import login

# Set page config for better display
st.set_page_config(page_title="LLaMA Chatbot", page_icon="🦙")
status_placeholder = st.empty()

# Set your HuggingFace token
try:
    hf_token = st.secrets["HF_TOKEN"]
    status_placeholder.success("🔑 Successfully found HF token in secrets!")
    login(token=hf_token)
    status_placeholder.success("🔑 Successfully logged in to Hugging Face!")
except Exception as e:
    status_placeholder.error(f"🚫 Error with HF token: {str(e)}")
    st.stop()

st.title("🦙 LLaMA Chatbot")

# Rest of the code...