Update functions.py

functions.py

@@ -54,6 +54,10 @@ from langchain.schema import (
 
 from langchain.prompts import PromptTemplate
 
+from langsmith import Client
+
+client = Client()
+
 nltk.download('punkt')
 
 
@@ -309,6 +313,10 @@ def inference(link, upload):
       
         return results, title
 
+@st.cache_resource
+def send_feedback(run_id, score):
+    client.create_feedback(run_id, "user_score", score=score)
+    
 @st.cache_data
 def clean_text(text):
     '''Clean all text after inference'''