Update app.py

app.py

@@ -51,22 +51,22 @@ llm_classes = {
 
 xml_system = """You're a helpful AI assistant. Given a user prompt and some related sources, \
 fulfill all the requirements of the prompt and provide citations. If a part of the generated text does \
-not use any of the sources, don't put a citation for that part. Otherwise, list all sources used for that part of the answer.
+not use any of the sources, don't put a citation for that part. Otherwise, list all sources used for that part of the text.
 At the end of each relevant part, add a citation in square brackets, numbered sequentially starting from [0], regardless of the source's original ID.
 
 
 Remember, you must return both the requested text and citations. A citation consists of a VERBATIM quote that \
-justifies the answer and a sequential number (starting from 0) for the quote's article. Return a citation for every quote across all articles \
-that justify the answer. Use the following format for your final output:
+justifies the text and a sequential number (starting from 0) for the quote's article. Return a citation for every quote across all articles \
+that justify the text. Use the following format for your final output:
 
-<cited_answer>
-    <answer></answer>
+<cited_text>
+    <text></text>
     <citations>
         <citation><source_id></source_id><source></source><quote></quote></citation>
         <citation><source_id></source_id><source></source><quote></quote></citation>
         ...
     </citations>
-</cited_answer>
+</cited_text>
 
 Here are the sources:{context}"""
 xml_prompt = ChatPromptTemplate.from_messages(
@@ -190,13 +190,13 @@ def generate_rag(
     result = rag_chain.invoke({"input": prompt})
     from pprint import pprint
     pprint(result)
-    return result['cited_answer'][0]['answer'], result['cited_answer'][1]['citations']
+    return result['cited_text'][0]['text'], result['cited_text'][1]['citations']
 
 
 def process_input(topic, length, tone, format_, pdfs):
     # Construct the prompt
     prompt = f"Write a {format_} about {topic} in about {length} words and a {tone} tone."
-    
+    print(prompt)
     # Generate the text and citations using RAG
     rag_output = generate_rag(
         prompt=prompt,