added citation in UI

.gitignore

@@ -0,0 +1 @@
+.env

app.py

@@ -1,72 +1,74 @@
 import gradio as gr
-from huggingface_hub import InferenceClient
+# from huggingface_hub import InferenceClient
+import requests
+from rich.pretty import pprint
+from langfuse import Langfuse
+from langfuse import observe
+import os
+from dotenv import load_dotenv
+load_dotenv()
 
+@observe()
+def respond(message, history):
+    url = "http://ec2-52-202-99-138.compute-1.amazonaws.com/api/v1/agents/6534b3ac82fa11f0b0019acfa5a3a049/completions"
+    headers = {
+        "Authorization": os.getenv("RAGFLOW_API_KEY"),
+        "Content-Type": "application/json"
+    }
+    payload = {
+    "question": message,
+    "stream": False,
+    }
+    response = requests.post(url, headers=headers, json=payload)
+    if response.status_code == 200:
+        agent_response = response.json()
+        pprint(agent_response)
+        citations = []
+        if agent_response["data"]["data"]["reference"]["chunks"]:
+            for id, chunk in agent_response["data"]["data"]["reference"]["chunks"].items():
+                citations.append({"cite_id":id, 
+                                  "content": chunk['content'],
+                                  "id": chunk["id"],
+                                  "document_name": chunk["document_name"],
+                                  "similarity": chunk["similarity"]})
 
-def respond(
-    message,
-    history: list[dict[str, str]],
-    system_message,
-    max_tokens,
-    temperature,
-    top_p,
-    hf_token: gr.OAuthToken,
-):
-    """
-    For more information on `huggingface_hub` Inference API support, please check the docs: https://huggingface.co/docs/huggingface_hub/v0.22.2/en/guides/inference
-    """
-    client = InferenceClient(token=hf_token.token, model="openai/gpt-oss-20b")
+          
+        history.append({"role": "assistant", "content": agent_response["data"]["data"]["content"]})
+        
+        if citations:
+            for cite in citations:
+                history.append({
+                    "role": "assistant",
+                    "content": (f"{cite['content']}"),
+                    "metadata": {"title": f"📚 Citation {cite['cite_id']}",
+                                #  "id": cite["id"],
+                                #  "parent_id": cite["document_name"],
+                                 "log": f" (Match: {100*cite['similarity']:.2f}%)"
+                                 }
+                })
+        return history
+    else:
+        return f"Error: {response.status_code} - {response.text}"
 
-    messages = [{"role": "system", "content": system_message}]
-
-    messages.extend(history)
-
-    messages.append({"role": "user", "content": message})
-
-    response = ""
-
-    for message in client.chat_completion(
-        messages,
-        max_tokens=max_tokens,
-        stream=True,
-        temperature=temperature,
-        top_p=top_p,
-    ):
-        choices = message.choices
-        token = ""
-        if len(choices) and choices[0].delta.content:
-            token = choices[0].delta.content
-
-        response += token
-        yield response
-
-
-"""
-For information on how to customize the ChatInterface, peruse the gradio docs: https://www.gradio.app/docs/chatinterface
-"""
 chatbot = gr.ChatInterface(
     respond,
     type="messages",
-    additional_inputs=[
-        gr.Textbox(value="You are a friendly Chatbot.", label="System message"),
-        gr.Slider(minimum=1, maximum=2048, value=512, step=1, label="Max new tokens"),
-        gr.Slider(minimum=0.1, maximum=4.0, value=0.7, step=0.1, label="Temperature"),
-        gr.Slider(
-            minimum=0.1,
-            maximum=1.0,
-            value=0.95,
-            step=0.05,
-            label="Top-p (nucleus sampling)",
-        ),
-    ],
 )
 
 with gr.Blocks() as demo:
-    with gr.Sidebar():
-        gr.LoginButton()
-    gr.load_chat("http://ec2-52-202-99-138.compute-1.amazonaws.com/api/v1/agents/6534b3ac82fa11f0b0019acfa5a3a049/completions", 
-                 model="EYSN_agent", token="Bearer ragflow-E5MDhhYTUwNzQ2OTExZjA5MjI0OWFjZm", streaming=False)
-    # chatbot.render()
+    # with gr.Sidebar():
+    #     gr.LoginButton()
+    gr.Markdown("# UNICEF EYSN Agent")
+    chatbot.render()
 
 
 if __name__ == "__main__":
-    demo.launch(debug=True)
+
+    langfuse = Langfuse(
+    secret_key=os.getenv("LANGFUSE_SECRET_KEY"),
+    public_key=os.getenv("LANGFUSE_PUBLIC_KEY"),
+    host="https://cloud.langfuse.com"
+    )
+
+    langfuse.track_event("demo.launch")
+    demo.launch(debug=True)