Update app.py

app.py

@@ -13,7 +13,8 @@ import openai
 load_dotenv()
 
 # Set OpenAI API key
-openai.api_key = os.environ['OPENAI_API_KEY']
+openai.api_key = os.environ("OPENAI_API_KEY")
+
 # Define the storage directory
 PERSIST_DIR = "./storage"
 
@@ -60,8 +61,12 @@ for message in st.session_state.messages:
         st.markdown(message["content"])
 
 # Get user input
+user_query_prefix = "Suggest all possible diagnosis, remedies and medicines for symptoms combining "
+
 if user_input := st.chat_input("Describe your symptoms or ask a medical question:"):
     # Add user message to chat history
+    user_input += user_query_prefix
+
     st.session_state.messages.append({"role": "user", "content": user_input})
     with st.chat_message("user"):
         st.markdown(user_input)