Update app.py

app.py

@@ -82,7 +82,7 @@ class LLM_Langchain():
         os.environ['HUGGINGFACEHUB_API_TOKEN'] = os.getenv("HF_KEY")
 
 
-    def generate_response(self, input_text):
+    def generate_response(self, input_text, context):
         
         template = f"<|system|>\nYou are a intelligent chatbot.</s>\n<|user|>\n{input_text}.\n<|assistant|>"
         llm = HuggingFaceHub(
@@ -96,7 +96,7 @@ class LLM_Langchain():
 
         result = llm_chain.run({
             "question": input_text,
-            # "context": context
+            "context": context
         })
         # return llm(input_text)
         return result
@@ -155,6 +155,11 @@ class LLM_Langchain():
                 for message in st.session_state.messages:
                     with st.chat_message(message.get('role')):
                         st.write(message.get("content"))
+
+                context = st.sidebar.text_input(
+                     label="Context",
+                     help="Context lets you know on what the answer should be generated"
+                 )
                 text = st.chat_input(disabled=text_input_visibility)
                 
                 if text:
@@ -173,7 +178,7 @@ class LLM_Langchain():
                     if self.selected_option == 'FineTune':
                         result = similarity(self.pdf_file_path, self.model_name, self.model_kwargs, text)
                     else:
-                        result = self.generate_response(text)
+                        result = self.generate_response(text, context)
                     st.session_state.messages.append(
                         {
                             "role": "assistant",