Update app.py

app.py

@@ -23,11 +23,15 @@ def get_sheet_order(file_path, 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):
-    # 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 combined prompt from selected values and weights
+def combine_selected_items(*selected_items_and_weights):
+    combined_prompt = []
+    for i in range(0, len(selected_items_and_weights), 2):
+        item = selected_items_and_weights[i]
+        weight = selected_items_and_weights[i+1]
+        if item:
+            combined_prompt.append(f"{item} (Weight: {weight})" if weight else item)
+    return f"Combined Prompt: {' | '.join(combined_prompt)}"
 
 # Function to create the Gradio interface
 def prompt_generator_interface(file_path, main_sheet_name):
@@ -36,22 +40,35 @@ def prompt_generator_interface(file_path, main_sheet_name):
 
     # 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}
 
+    # Define the sheets that should NOT have weights
+    no_weight_sheets = ["Poetic", "Scenarios", "Camera Setup", "Combos", "PositiceNegative prompts", "LAZY Mode"]
+
+    # Move Resources tab to the last place
+    if 'Resources' in valid_sheets:
+        valid_sheets['Resources'] = valid_sheets.pop('Resources')
+
     # Gradio interface
     with gr.Blocks() as interface:
         gr.Markdown("# 📝 Witness Prompt Generator\nSelect an item from each sheet to generate a combined prompt.")
 
-        # Initialize an empty list to store all the dropdowns and user selections
-        dropdowns = []
-        
+        # Initialize an empty list to store all the dropdowns and weight inputs
+        dropdowns_and_weights = []
+
         # Display all sheets as dropdowns in the order specified by the main sheet
         with gr.Row():
             for sheet_name in valid_sheets:
-                dropdown = gr.Dropdown(choices=valid_sheets[sheet_name], label=sheet_name, interactive=True)
-                dropdowns.append(dropdown)
+                with gr.Column():
+                    dropdown = gr.Dropdown(choices=valid_sheets[sheet_name], label=sheet_name, interactive=True)
+                    dropdowns_and_weights.append(dropdown)
+                    
+                    # If the sheet is not in the no_weight_sheets, add a weight input
+                    if sheet_name not in no_weight_sheets:
+                        weight_input = gr.Textbox(label=f"{sheet_name} Weight", placeholder="Enter weight (optional)", interactive=True)
+                        dropdowns_and_weights.append(weight_input)
 
         # Button to submit the combined prompt
         submit_button = gr.Button("Generate Combined Prompt")
@@ -60,7 +77,7 @@ def prompt_generator_interface(file_path, main_sheet_name):
         combined_output = gr.Textbox(label="Combined Prompt", placeholder="Your combined prompt will appear here...")
 
         # Action when the submit button is clicked
-        submit_button.click(combine_selected_items, inputs=dropdowns, outputs=combined_output)
+        submit_button.click(combine_selected_items, inputs=dropdowns_and_weights, outputs=combined_output)
 
     return interface