Update app.py

app.py

@@ -1035,7 +1035,15 @@ try:
         api_key = "sk-1a0127fbc52a4e50a93bd5cd18af3a85"
         prompt_user = st.text_input("Write a brief description of the selected column question.")
         if st.button("Submit"):
-            df2 = categorize_responses(df[["id"]+selected_list], api_key, prompt=prompt_user)
+
+            df2 = categorize_responses(initial_prompt = prompt_user, 
+                        dataframe = df[["id"]+selected_list],
+                        id_column = "id",
+                        text_column = selected_list[0],
+                        api_key = api_key,
+                        max_retries: int = 3,
+                        delay: float = 1.0)
+            
             st.subheader("Categorized data")
             st.dataframe(df2)