Update app.py

app.py

@@ -1,5 +1,6 @@
 import pandas as pd
 import gradio as gr
+from ui_components import CustomDropdown, CustomButton  # Example: Replace with actual component names
 
 # Function to load Excel sheets and extract first column data from valid sheets
 def load_excel_sheets(file_path):
@@ -51,7 +52,7 @@ def prompt_generator_interface(file_path, main_sheet_name):
     if 'Resources' in valid_sheets:
         valid_sheets['Resources'] = valid_sheets.pop('Resources')
 
-    # Gradio interface
+    # Gradio interface using custom UI components
     with gr.Blocks() as interface:
         gr.Markdown("# 📝 Witness Prompt Generator\nSelect an item from each sheet to generate a combined prompt.")
 
@@ -62,16 +63,18 @@ def prompt_generator_interface(file_path, main_sheet_name):
         with gr.Row():
             for sheet_name in valid_sheets:
                 with gr.Column():
-                    dropdown = gr.Dropdown(choices=valid_sheets[sheet_name], label=sheet_name, interactive=True)
+                    # Using the custom dropdown from ui_components.py
+                    dropdown = CustomDropdown(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:
+                        # Custom weight input
                         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")
+        # Using a custom button for submission
+        submit_button = CustomButton("Generate Combined Prompt")
 
         # Textbox to display the combined prompt
         combined_output = gr.Textbox(label="Combined Prompt", placeholder="Your combined prompt will appear here...")