Update pages/Phase1.py

pages/Phase1.py

@@ -21,11 +21,11 @@ model_info = {
     }
 }
 
-def format_prompt(message, custom_instructions=None):
+def format_prompt(context, question, custom_instructions=None):
     prompt = ""
     if custom_instructions:
         prompt += f"[INST] {custom_instructions} [/INST]"
-    prompt += f"[INST] {message} [/INST]"
+    prompt += f"{context}\n\n[INST] {question} [/INST]"
     return prompt
 
 def reset_conversation():
@@ -99,9 +99,6 @@ if uploaded_file is not None:
 if prompt := st.chat_input(f"Hi I'm {selected_model}, ask me a question"):
 
     custom_instruction = "Act like a Human in conversation"
-    
-    if "pdf_text" in st.session_state:
-        prompt = f"{st.session_state.pdf_text}\n\nQuestion: {prompt}"
 
     # Display user message in chat message container
     with st.chat_message("user"):
@@ -109,7 +106,8 @@ if prompt := st.chat_input(f"Hi I'm {selected_model}, ask me a question"):
     # Add user message to chat history
     st.session_state.messages.append({"role": "user", "content": prompt})
 
-    formated_text = format_prompt(prompt, custom_instruction)
+    context = st.session_state.pdf_text if "pdf_text" in st.session_state else ""
+    formated_text = format_prompt(context, prompt, custom_instruction)
 
     # Display assistant response in chat message container
     with st.chat_message("assistant"):