Update app.py

app.py

@@ -156,13 +156,20 @@ def process_inputs(llm, file, relevance, diversity):
         # Extract the first column
         questions_list = df.iloc[:, 0]
 
-        # Generate prompts for each question
-        prompts_list = questions_list.apply(generate_prompts)
+        # Initialize lists to store the expanded data
+        expanded_questions = []
+        expanded_prompts = []
 
-        # Combine the questions and prompts into a DataFrame
+        # Generate prompts for each question and expand the data
+        for question in questions_list:
+            prompts = generate_prompts(question)
+            expanded_questions.extend([question] * len(prompts))
+            expanded_prompts.extend(prompts)
+
+        # Combine the expanded data into a DataFrame
         output_df = pd.DataFrame({
-            'Questions': questions_list,
-            'Generated Prompts': prompts_list
+            'Questions': expanded_questions,
+            'Generated Prompts': expanded_prompts
         })
 
         # Save the DataFrame to a new Excel file