Update app.py

app.py

@@ -9,21 +9,69 @@ from huggingface_hub import login
 st.set_page_config(page_title="LLaMA Chatbot", page_icon="🦙")
 status_placeholder = st.empty()
 
-# Try to get the token from public variable
+# Authentication (keeping the working code)
 try:
-    # Note we changed HF_TOKEN25 to HF_TOKEN to match your variable name
     hf_token = os.environ.get("HF_TOKEN")
-    
     if not hf_token:
         raise ValueError("Token not found")
-        
-    status_placeholder.success("🔑 Successfully found HF token!")
     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.error("Please ensure the token is properly set as a public variable")
     st.stop()
 
-st.title("🦙 LLaMA Chatbot")
+st.title("🦙 LLaMA Chatbot")
+
+# Model loading with detailed status updates
+@st.cache_resource
+def load_model():
+    try:
+        model_path = "Alaaeldin/llama2-app"
+        
+        with st.spinner("🔄 Loading tokenizer..."):
+            tokenizer = AutoTokenizer.from_pretrained(
+                model_path,
+                token=hf_token,
+                trust_remote_code=True
+            )
+            st.success("✅ Tokenizer loaded!")
+        
+        with st.spinner("🔄 Loading model... This might take a few minutes..."):
+            model = AutoModelForCausalLM.from_pretrained(
+                model_path,
+                torch_dtype=torch.float16,
+                device_map="auto",
+                load_in_8bit=True,
+                token=hf_token,
+                trust_remote_code=True
+            )
+            st.success("✅ Model loaded!")
+        
+        return model, tokenizer
+    except Exception as e:
+        st.error(f"❌ Error loading model: {str(e)}")
+        return None, None
+
+# Initialize chat history
+if "messages" not in st.session_state:
+    st.session_state.messages = []
+
+# Load model
+model, tokenizer = load_model()
+
+# Display chat interface
+if model and tokenizer:
+    st.success("✨ Ready to chat! Enter your message below.")
+    
+    # Display chat history
+    for message in st.session_state.messages:
+        with st.chat_message(message["role"]):
+            st.markdown(message["content"])
+
+    # Chat input
+    if prompt := st.chat_input("Your message"):
+        # Add user message to chat history
+        st.session_state.messages.append({"role": "user", "content": prompt})
+        # Display user message
+        with st.chat_message("user"):
+            st.markdown(prompt)