Update app.py

app.py

@@ -1,7 +1,5 @@
-# 1. Import necessary libraries
 import pandas as pd
 import gradio as gr
-import random
 
 # Function to load Excel sheets and extract first column data from valid sheets
 def load_excel_sheets(file_path):
@@ -13,42 +11,47 @@ def load_excel_sheets(file_path):
         df = pd.read_excel(xls, sheet_name=sheet_name)
         # Check if the sheet has at least one column and is not empty
         if not df.empty and df.shape[1] > 0:
-            # Extract the first column and remove NaN values
+            # Extract the first column and keep the order intact
             sheets[sheet_name] = df.iloc[:, 0].dropna().tolist()
-        else:
-            # If the sheet is empty or has no valid data, skip it
-            sheets[sheet_name] = []  # Optional: You can skip or keep an empty list
     
     return sheets
 
-# Function to generate a random item from the selected sheet
-def generate_random_item(sheet_name, sheets):
-    items = sheets.get(sheet_name, [])
-    if items:
-        return random.choice(items)
-    return "No items available."
+# 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}"
 
-# Gradio interface
+# Function to create the Gradio interface
 def prompt_generator_interface(file_path):
     # Load all the sheets and extract data from the first column
     sheets = load_excel_sheets(file_path)
 
-    # Define function for Gradio
-    def generate_prompt(sheet_name):
-        return generate_random_item(sheet_name, sheets)
-
-    # Improved UI with better layout
-    interface = gr.Interface(
-        fn=generate_prompt,  # Function to call for generating random item
-        inputs=gr.Dropdown(choices=list(sheets.keys()), label="Select Sheet"),  # Updated input syntax
-        outputs=gr.Textbox(label="Generated Prompt"),  # Updated output syntax
-        title="📝 Witness Prompt Generator",
-        description="Select a sheet to generate a random prompt from the provided data.",
-        theme="default",  # Apply a clean default theme
-        live=False,  # Ensure that generation only happens on button click
-        css=".gradio-container {background-color: #f5f5f5; font-family: Arial; padding: 20px;} h1 {color: #333;}"
-    )
-    
+    # 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 = []
+        
+        # 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
+        with gr.Row():
+            for sheet_name in reversed_sheet_names:
+                dropdown = gr.Dropdown(choices=sheets[sheet_name], label=sheet_name, interactive=True)
+                dropdowns.append(dropdown)
+
+        # Button to submit the combined prompt
+        submit_button = gr.Button("Generate Combined Prompt")
+
+        # Textbox to display the combined prompt
+        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)
+
     return interface
 
 # Create and launch the Gradio interface