Update app.py

app.py

@@ -10,7 +10,6 @@ from langchain.chains.combine_documents import create_stuff_documents_chain
 from langchain_core.prompts import ChatPromptTemplate
 from dotenv import load_dotenv
 from src.prompt import system_prompt
-import base64
 
 # Set up cache directories
 os.environ['TRANSFORMERS_CACHE'] = '/tmp/model_cache'
@@ -202,7 +201,8 @@ init_rate_limiting()
 
 # Sidebar content
 with st.sidebar:
-    st.image("https://online.flipbuilder.com/clinical-library/vxes/files/shot.png", width=200)
+    # Using a more reliable image URL - or you can upload an image to your Space
+    st.image("https://img.freepik.com/free-vector/online-doctor-concept_23-2148522199.jpg", width=200)
     st.title("MedPrep AI")
     
     st.markdown("### About")
@@ -269,7 +269,7 @@ def initialize_rag_chain():
             
             retriever = docsearch.as_retriever(search_type="similarity", search_kwargs={"k": 3})
             
-            llm = OpenAI(temperature=0.4, max_tokens=500)
+            llm = OpenAI(model="gpt-3.5-turbo", temperature=0.4, max_tokens=500)
             
             prompt = ChatPromptTemplate.from_messages([
                 ("system", system_prompt),
@@ -351,37 +351,21 @@ with st.container():
                     try:
                         result = rag_chain.invoke({"input": prompt})
                         response = result.get("answer", "Sorry, I couldn't find an answer to that in First Aid.")
-                        remaining = MAX_REQUESTS_PER_DAY - count
-                        response_with_footer = response
                         
                         # Display response without footer
                         display_message(response)
                         
-                        # Update sidebar with new remaining queries
-                        remaining_queries = get_remaining_queries(user_id)
-                        used_queries = MAX_REQUESTS_PER_DAY - remaining_queries
-                        percentage = (used_queries / MAX_REQUESTS_PER_DAY) * 100
-                        
-                        # Need to rerun to update the sidebar
-                        st.sidebar.markdown("### Daily Query Limit")
-                        st.sidebar.markdown(
-                            f"""
-                            <div class="progress-container">
-                                <div class="progress-bar" style="width: {percentage}%;"></div>
-                            </div>
-                            <p class="queries-text">{remaining_queries} of {MAX_REQUESTS_PER_DAY} queries remaining today</p>
-                            """,
-                            unsafe_allow_html=True
-                        )
-                        
                     except Exception as e:
-                        response_with_footer = f"Error processing your request: {str(e)}"
-                        display_message(response_with_footer)
+                        error_response = f"Error processing your request: {str(e)}"
+                        display_message(error_response)
+                        st.session_state.messages.append({"role": "assistant", "content": error_response})
+                        st.experimental_rerun()
                 
                 # Add assistant response to chat history
                 st.session_state.messages.append({"role": "assistant", "content": response})
             
-            # Clear the input box
+            # Clear the input box - using a different approach that's more reliable
+            st.session_state.user_input = ""
             st.experimental_rerun()
 
 # Footer