Add Weave

app.py

@@ -23,6 +23,7 @@ from langchain_text_splitters import RecursiveCharacterTextSplitter
 from langchain_core.vectorstores import VectorStoreRetriever
 
 import time 
+import weave
 
 from str2escaped import str2escaped
 
@@ -441,6 +442,7 @@ def get_json(text: str):
     return json.loads(text)
 
 
+@weave.op
 def detect_features(image_paths, garment_type, language="English"):
     # Path to your image
     # image_path = "path_to_your_image.jpg"
@@ -517,6 +519,7 @@ def parse_structure(struct_ref):
     return types, languages
 
 
+@weave.op
 def detect_language(texts, model):
     langs = ["_"] * len(texts)
 
@@ -640,6 +643,7 @@ def evaluate(descriptions,
     return response
 
 
+@weave.op
 def generate(*data):
     global visible
     print("visible", visible)
@@ -901,6 +905,9 @@ def authf(username, password):
 
 if __name__ == '__main__':
     # demo.launch(server_name="0.0.0.0", auth=authf)
+
+    weave.init("Copywriting - Hugging Face")
+
     with open("formatted_prompts.txt", "w") as f:
         f.writelines('"write_struct_copy": "' + str2escaped(struct_copy_prompt).strip() + '",\n')
         f.writelines('"write_copy": "' + str2escaped(copy_prompt).strip() + '",\n')

requirements.txt

@@ -9,4 +9,5 @@ langchain_chroma
 langchainhub
 langchain_community
 langchain-google-genai
-google-generativeai
+google-generativeai
+weave