Update app.py

app.py

@@ -39,15 +39,19 @@ if assistant_api_key == '':
 else:
     gpt_assistant = GPTAssistant(assistant_api_key)
 
-
-html_content = st.text_input("Paste the HTML tags of the item you want to extract:", max_chars=10000, help="example: <li>Product 1 </li>, watch the video above")
+html_content = None
 # check if html_content is an url, and show error if it is
-if html_content:
-    if html_content.startswith("http"):
-        st.write("Please paste the HTML piece code, not the URL")
-        html_content = None
+@traceable(run_type="tool")
+def html_content_input:
+    html_content = st.text_input("Paste the HTML tags of the item you want to extract:", max_chars=10000, help="example: <li>Product 1 </li>, watch the video above")
+        if html_content:
+            if html_content.startswith("http"):
+                st.write("Please paste the HTML piece code, not the URL")
+                html_content = None
+        
+        return st.button("Generate output format & code")
 
-extract_button = st.button("Generate output format & code")
+extract_button = html_content_input()
 
 if html_content and extract_button:
     try: