Update app.py

app.py

@@ -547,7 +547,7 @@ def categorize_responses(df, api_key, prompt=None):
     Categorize responses in a DataFrame using OpenAI's GPT-based API.
 
     Args:
-        df (pd.DataFrame): Input DataFrame with columns 'id' and 'response'.
+        df (pd.DataFrame): Input DataFrame with columns 'id' and 'p1'.
         api_key (str): OpenAI API key for accessing GPT-based models.
         prompt (str, optional): A question or explanation of the survey question for better categorization.
 
@@ -566,7 +566,7 @@ def categorize_responses(df, api_key, prompt=None):
     categories = []
 
     # Iterate through the responses in the DataFrame
-    for response in df.drop(columns=['id']):
+    for response in df['p1']:
         # Construct the prompt for the specific response
         full_prompt = f"{base_prompt}\n\nResponse: {response}\n\nCategory:"