Update app.py

app.py

@@ -1,7 +1,7 @@
 import streamlit as st
 from llama_cpp import Llama
 
-st.set_page_config(page_title="Chat with AI", page_icon="🤖")
+st.set_page_config(page_title="Chat with AI", page_icon="🤖", layout="wide")
 
 # Custom CSS for better styling
 st.markdown("""
@@ -57,6 +57,36 @@ def format_context(messages):
             context += f"Assistant: {message['content']}\n"
     return context
 
+# Sidebar
+st.sidebar.title("Chat with AI")
+st.sidebar.markdown("This is a simple chat interface using Streamlit and an AI model.")
+
+# Add useful information to the sidebar
+st.sidebar.header("How to use")
+st.sidebar.markdown("""
+1. Type your question in the chat input box at the bottom of the screen.
+2. Press Enter or click the Send button to submit your question.
+3. The AI will generate a response based on your input.
+4. You can have a continuous conversation by asking follow-up questions.
+""")
+
+st.sidebar.header("Model Information")
+st.sidebar.markdown("""
+- Model: med_phi3-mini-4k-GGUF
+- Context Length: 512 tokens
+- This model is specialized in medical knowledge.
+""")
+
+st.sidebar.header("Tips")
+st.sidebar.markdown("""
+- Be clear and specific in your questions.
+- For medical queries, provide relevant details.
+- Remember that this is an AI model and may not always be 100% accurate.
+""")
+
+# Main chat interface
+st.title("Chat with AI")
+
 # Initialize chat history
 if "messages" not in st.session_state:
     st.session_state.messages = []
@@ -101,5 +131,14 @@ if prompt := st.chat_input("What is your question?"):
     # Add assistant response to chat history
     st.session_state.messages.append({"role": "assistant", "content": assistant_response})
 
-st.sidebar.title("Chat with AI")
-st.sidebar.markdown("This is a simple chat interface using Streamlit and an AI model.")
+# Add a button to clear the chat history
+if st.sidebar.button("Clear Chat History"):
+    st.session_state.messages = []
+    st.experimental_rerun()
+
+# Display the number of messages in the current conversation
+st.sidebar.markdown(f"Current conversation length: {len(st.session_state.messages)} messages")
+
+# Add a footer
+st.sidebar.markdown("---")
+st.sidebar.markdown("Created with ❤️ using Streamlit and Llama.cpp")