Update app.py

app.py

@@ -3,7 +3,7 @@ import requests
 import streamlit as st
 from llama_cpp import Llama
 
-# ✅ Set Streamlit Page Config (Must be First)
+# ✅ Streamlit Page Config (Must be first)
 st.set_page_config(page_title="Phi-3 Mini Chatbot", layout="centered")
 
 # ✅ Define model path
@@ -25,8 +25,13 @@ if not os.path.exists(MODEL_PATH):
         st.stop()
 
 # ✅ Load model
-if "model" not in st.session_state:
-    st.session_state["model"] = Llama(model_path=MODEL_PATH, n_ctx=4096)
+try:
+    if "model" not in st.session_state:
+        st.session_state["model"] = Llama(model_path=MODEL_PATH, n_ctx=4096)
+        st.write("✅ Model loaded successfully!")
+except Exception as e:
+    st.error(f"🚨 Error loading model: {e}")
+    st.stop()
 
 # Streamlit UI setup
 st.title("🤖 Phi-3 Mini Chatbot")
@@ -52,13 +57,15 @@ if st.button("Send") and user_input:
     st.chat_message("user").write(user_input)
     
     # Generate response
-    response = st.session_state["model"].create_completion(
-        prompt=user_input, max_tokens=1024, temperature=0.7, top_p=0.9
-    )["choices"][0]["text"].strip()
-    
-    # Add model response to chat history
-    st.session_state["messages"].append(("assistant", response))
-    st.chat_message("assistant").write(response)
+    try:
+        response = st.session_state["model"].create_completion(
+            prompt=user_input, max_tokens=1024, temperature=0.7, top_p=0.9
+        )
+        response_text = response["choices"][0]["text"].strip()
+        st.session_state["messages"].append(("assistant", response_text))
+        st.chat_message("assistant").write(response_text)
+    except Exception as e:
+        st.error(f"🚨 Error generating response: {e}")
 
 # Run the app with: streamlit run app.py