Update app.py

app.py

@@ -949,10 +949,11 @@ try:
             cols,
             default=[]
         )
+        df["id"] = df.index
         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[selected_list], api_key, prompt=prompt_user)
+            df2 = categorize_responses(df[["id"]+selected_list], api_key, prompt=prompt_user)
             st.subheader("Categorized data")
             st.dataframe(df2)