Update app.py

app.py

@@ -10,6 +10,7 @@ from huggingface_hub import hf_hub_download
 from huggingface_hub import HfApi, login
 from datetime import datetime
 from langchain_openai import ChatOpenAI
+from langchain_community.callbacks import get_openai_callback
 
 
 from langchain_core.prompts import ChatPromptTemplate, PromptTemplate
@@ -108,10 +109,12 @@ def respond(message, history):
         ]
         return "", new_history
     else:
-        answer = qa_chain.invoke(message)
+        with get_openai_callback() as cb:
+            answer = qa_chain.invoke({"input": message})
+        #answer = qa_chain.invoke(message)
         docs = retriever.invoke(message)
         refs = [f"Page {d.metadata.get('page', 'N/A')}" for d in docs]
-        full_answer = f"{answer}\n\n**References:**\n" + "\n".join(refs)
+        full_answer = f"Input tokens: {cb.prompt_tokens}\n Completion tokens: {cb.completion_tokens}\nTotal tokens: {cb.total_tokens}\n\n{answer}\n\n**References:**\n" + "\n".join(refs)
         
         # CRITICAL: Append ONLY pure dicts - no metadata, tuples, or extras
         new_history = history + [  # Or history.append() then return history
@@ -411,6 +414,12 @@ with gr.Blocks(css=css) as demo:
                     file_types=[".pdf"],
                     elem_id="oem-import-file"
                 )
+                manabLebel=gr.HTML("""
+                        <div style="color: red; background: black; padding: 20px; text-align: justify; font-size: 20px;">
+                            Disclaimer: AI generated outcome is only for reference. Consider the 
+                            following only as a draft and recheck is mandatory.
+                        </div>"""                        
+                )
             run_btn_oemImport = gr.Button("Generate CQ of OEM-offer", elem_id="generateOEMImport-btn")
             out_tech = gr.HTML(label="Generated CQ")
             #out = gr.Textbox(lines=15, label="Compliance Result",elem_id="compliance-out")