Update app.py

app.py

@@ -94,23 +94,6 @@ st.set_page_config(
     layout="centered"
 )
 
-# Custom CSS to style images and improve layout
-st.markdown("""
-<style>
-.header-container {
-    display: flex;
-    align-items: center;
-    gap: 20px;
-}
-.header-image {
-    width: 80px;
-}
-.sidebar-image {
-    margin-bottom: 20px;
-}
-</style>
-""", unsafe_allow_html=True)
-
 # Initialize session state for chat history
 if 'messages' not in st.session_state:
     st.session_state.messages = []
@@ -118,25 +101,10 @@ if 'messages' not in st.session_state:
 # Initialize rate limiting
 init_rate_limiting()
 
-# Add image to sidebar
-with st.sidebar:
-    st.image("https://img.freepik.com/free-vector/medical-education-abstract-concept-vector-illustration_107173-26967.jpg", width=200, caption="First Aid USMLE", use_column_width=True)
-    st.markdown("#### About")
-    st.write("This AI assistant is powered by First Aid USMLE Step 1 content and helps medical students prepare for their exams.")
-    
-    # Display remaining queries
-    user_id = get_user_id()
-    remaining_queries = get_remaining_queries(user_id)
-    st.write(f"Remaining queries today: {remaining_queries}/{MAX_REQUESTS_PER_DAY}")
-    
-    st.markdown("---")
-    st.markdown("### Tips for Better Results")
-    st.markdown("""
-    - Be specific with your questions
-    - Include relevant keywords
-    - Ask one concept at a time
-    - Mention the system or topic area
-    """)
+# Display remaining queries
+user_id = get_user_id()
+remaining_queries = get_remaining_queries(user_id)
+st.sidebar.write(f"Remaining queries today: {remaining_queries}/{MAX_REQUESTS_PER_DAY}")
 
 # Check for API keys
 PINECONE_API_KEY = os.environ.get('PINECONE_API_KEY')
@@ -166,7 +134,7 @@ def initialize_rag_chain():
         retriever = docsearch.as_retriever(search_type="similarity", search_kwargs={"k": 3})
         
         st.sidebar.write("Initializing OpenAI...")
-        llm = OpenAI(model="gpt-3.5-turbo", temperature=0.4, max_tokens=500)
+        llm = OpenAI(temperature=0.4, max_tokens=500)
         
         prompt = ChatPromptTemplate.from_messages([
             ("system", system_prompt),
@@ -184,14 +152,8 @@ def initialize_rag_chain():
         st.sidebar.text(traceback.format_exc())
         return None
 
-# Main app header with logo
-st.markdown("""
-<div class="header-container">
-    <img src="https://med.stanford.edu/content/dam/sm-news/images/2014/09/first-aid-logo.gif" class="header-image">
-    <h1>First AID USMLE Step 1</h1>
-</div>
-""", unsafe_allow_html=True)
-
+# Main app title
+st.title("First AID USMLE Step 1")
 st.write("Ask me any question from First AID USMLE Step 1 book, and I'll try to help!")
 
 # Initialize the RAG chain
@@ -240,14 +202,5 @@ if prompt := st.chat_input("Ask a question..."):
 
 # Footer
 st.markdown("---")
-# Add a small image in the footer
-cols = st.columns([1, 2, 1])
-with cols[1]:
-    st.image("https://i.pinimg.com/originals/ec/81/1c/ec811c4588d10f8286aad7e3ed33e478.png", width=80)
-
-st.markdown("""
-<div style="text-align: center; margin-top: 10px;">
-    <p style="font-size: 0.8rem;">This AI assistant uses retrieval augmented generation to provide information from First Aid USMLE Step 1 content.
-    It's designed to help with studying, but should not replace professional medical advice.</p>
-</div>
-""", unsafe_allow_html=True)
+st.markdown("*This AI assistant uses retrieval augmented generation to provide information from First Aid USMLE Step 1 content.
+        It's designed to help with studying, but should not replace professional medical advice..*")