Update app.py

app.py

@@ -84,18 +84,15 @@ class DocumentProcessor:
         return self.vectordb_doc
 
 
-    def parse_output(self, content):
-        prompt = ChatPromptTemplate.from_messages([
-            ("system", "You are an expert content parser and your task is convert the content into json format"),
-            ("user", "{input}")
-        ])
-
-        # chain = LLMChain(llm=self.llm, prompt=message)
-        chain = prompt | self.llm
-        result = chain.invoke(content)
-
-        # result = self.llm.invoke(message)
-        return result
+    def parse_output(self, response):
+        # Find the index where "Question:" starts
+        question_index = response.find("Question:")
+        # Get all text including and after "Question:"
+        if question_index != -1:
+            result_text = response[question_index:].strip()
+            return result_text
+        else:
+            return "I apologies, I don't know the answer"
 
     def document_chain(self):
         # prompt = ChatPromptTemplate.from_template("""Answer the following question based only on the provided context:
@@ -123,7 +120,7 @@ class DocumentProcessor:
     def get_response(self, retrieval_chain, message):
         # response = retrieval_chain.invoke({"input": "how can langsmith help with testing?"})
         response = retrieval_chain.invoke({"input": message})
-        print(response["answer"])
+        # print(response["answer"])
         return response["answer"]
 
 
@@ -155,7 +152,7 @@ def echo(message, history, processor):
 
         chain_result = processor.get_response(reterival_chain, message)
         parsed_result = processor.parse_output(chain_result)
-        return chain_result
+        return parsed_result
 
     except Exception as e:
         # Handle any exceptions that occur during execution
@@ -203,9 +200,10 @@ def main():
                 def process_echo(message, history):
                     return echo(message, history, processor)
 
-                gr.ChatInterface(fn=process_echo, examples=["what is title", "what is summary", "merhaba"])
+                gr.ChatInterface(fn=process_echo, examples=["what is title", "what is summary", "create notes"])
+
+                gr.Markdown("* Note: The answers can be incorrect due to Smaller Model/AI")
 
-                gr.Markdown("[Note: The answers can be incorrect as we are using smaller model]")
 
     # with gr.Blocks() as demo:
     #         gr.ChatInterface(fn=echo, examples=["what is title", "what is summary", "merhaba"], title="chat with your data")