Update app.py

app.py

@@ -1,19 +1,24 @@
 import streamlit as st
 import ollama
 
+def get_llama_response(user_input):
+    try:
+        response = ollama.chat(model='llama2', messages=[{'role': 'user', 'content': user_input}])
+        return response['message']['content']
+    except Exception as e:
+        st.error(f"An error occurred: {str(e)}")
+        return "Sorry, I couldn't process your request at the moment. Please try again later."
+
 def main():
     st.title("Llama Chatbot")
-    
-    user_input = st.text_input("You:", "Why is the sky blue?")
+
+    user_input = st.text_area("You:", placeholder="Enter your message here...")
 
     if st.button("Ask"):
         with st.spinner("Thinking..."):
-            response = ollama.chat(model='llama2', messages=[{
-                'role': 'user',
-                'content': user_input,
-            }])
+            response = get_llama_response(user_input)
         st.success("Llama says:")
-        st.write(response['message']['content'])
+        st.write(response)
 
 if __name__ == "__main__":
-    main()
+    main()