Update app.py

app.py

@@ -16,17 +16,30 @@ def load_excel_sheets(file_path):
     
     return sheets
 
+# Function to get the order of sheets from the main sheet
+def get_sheet_order(file_path, main_sheet_name):
+    # Read the main sheet which contains the sheet order
+    df_order = pd.read_excel(file_path, sheet_name=main_sheet_name)
+    # Assuming the sheet names are listed in the first column
+    return df_order.iloc[:, 0].dropna().tolist()
+
 # Function to create the combined prompt from selected values
-def combine_selected_items(selected_items):
+def combine_selected_items(*selected_items):
     # Combine all the selected items into a single prompt
     combined_prompt = " | ".join([item for item in selected_items if item])
     return f"Combined Prompt: {combined_prompt}"
 
 # Function to create the Gradio interface
-def prompt_generator_interface(file_path):
+def prompt_generator_interface(file_path, main_sheet_name):
     # Load all the sheets and extract data from the first column
     sheets = load_excel_sheets(file_path)
 
+    # Get the sheet order from the main sheet
+    sheet_order = get_sheet_order(file_path, main_sheet_name)
+    
+    # Filter out sheets that don't exist in the main order
+    valid_sheets = {sheet_name: sheets[sheet_name] for sheet_name in sheet_order if sheet_name in sheets}
+
     # Gradio interface
     with gr.Blocks() as interface:
         gr.Markdown("# 📝 Witness Prompt Generator\nSelect an item from each sheet to generate a combined prompt.")
@@ -34,13 +47,10 @@ def prompt_generator_interface(file_path):
         # Initialize an empty list to store all the dropdowns and user selections
         dropdowns = []
         
-        # Reverse the order of sheets to display them right to left
-        reversed_sheet_names = list(sheets.keys())[::-1]
-        
-        # Display all sheets as dropdowns in a grid
+        # Display all sheets as dropdowns in the order specified by the main sheet
         with gr.Row():
-            for sheet_name in reversed_sheet_names:
-                dropdown = gr.Dropdown(choices=sheets[sheet_name], label=sheet_name, interactive=True)
+            for sheet_name in valid_sheets:
+                dropdown = gr.Dropdown(choices=valid_sheets[sheet_name], label=sheet_name, interactive=True)
                 dropdowns.append(dropdown)
 
         # Button to submit the combined prompt
@@ -59,6 +69,9 @@ if __name__ == "__main__":
     # Path to your Excel file (adjust the path if necessary)
     file_path = 'Witness Prompt Generator.xlsm'
     
+    # Define the name of the main sheet that contains the order of sheets
+    main_sheet_name = 'Main Order'  # Change this to match the name of your control sheet
+    
     # Create and launch interface
-    interface = prompt_generator_interface(file_path)
+    interface = prompt_generator_interface(file_path, main_sheet_name)
     interface.launch()