Update app.py

app.py

@@ -55,6 +55,7 @@ def predict_return(selected_products, total_customer_purchases, total_customer_r
     except requests.exceptions.RequestException as e:
         return f"Error: {str(e)}", ""
 
+
 # Predefined list of model-fabric-color combinations
 combinations = [
     "01CA9T-0130C-922",
@@ -65,43 +66,41 @@ combinations = [
 ]
 
 # Gradio interface elements
-inventory_column = gr.Column(
-    [
-        gr.CheckboxGroup(choices=combinations, label="Select Products", type="value"),  # Scrollable inventory
-    ]
-)
+inventory_checkbox_group = gr.CheckboxGroup(choices=combinations, label="Select Products", type="value")
 
-user_info_column = gr.Column(
-    [
-        gr.Textbox(value="User Information\nTotal Purchases: 0\nTotal Returns: 0", label="User Info", interactive=False),
-    ]
-)
+# Slider elements for total purchases and returns
+total_purchases_slider = gr.Slider(0, 10, step=1, label="Total Customer Purchases", value=0)
+total_returns_slider = gr.Slider(0, 10, step=1, label="Total Customer Returns", value=0)
 
-cart_column = gr.Column(
-    [
-        gr.Textbox(value="", label="Cart", interactive=False),  # Cart details
-        gr.Textbox(value="", label="Prediction Results", interactive=False),  # Predictions for selected products
-    ]
-)
+# Output elements for predictions and cart details
+cart_output = gr.Textbox(value="", label="Cart", interactive=False)
+predictions_output = gr.Textbox(value="", label="Prediction Results", interactive=False)
+
+# User information output
+user_info_output = gr.Textbox(value="User Information\nTotal Purchases: 0\nTotal Returns: 0", label="User Info", interactive=False)
 
 # Layout with two main columns: Left (Inventory) and Right (User Info + Cart)
 with gr.Row():
-    gr.Column([inventory_column])  # Left side: Inventory
     with gr.Column():
-        gr.Column([user_info_column])  # Right side: User Info
-        gr.Column([cart_column])  # Right side: Cart & Predictions
+        inventory_checkbox_group  # Left side: Inventory
+    with gr.Column():
+        user_info_output  # Right side: User Info
+        cart_output  # Right side: Cart & Predictions
+        predictions_output  # Right side: Prediction Results
+
 
 # Gradio Interface
 interface = gr.Interface(
     fn=predict_return,  # Function to process predictions
     inputs=[
-        inventory_column,  # Left side: Inventory
-        gr.Slider(0, 10, step=1, label="Total Customer Purchases", value=0),  # Total purchases
-        gr.Slider(0, 10, step=1, label="Total Customer Returns", value=0)  # Total returns
+        inventory_checkbox_group,  # Left side: Inventory
+        total_purchases_slider,  # Total purchases
+        total_returns_slider  # Total returns
     ],
     outputs=[
-        cart_column,  # Right side: Cart & Predictions
-        user_info_column,  # Right side: User Info
+        predictions_output,  # Right side: Cart & Predictions
+        user_info_output,  # Right side: User Info
+        cart_output  # Right side: Cart
     ],
     live=True  # Enable live interaction
 )