Update app.py

app.py

@@ -54,15 +54,14 @@ def on_predict(row_label):
     error_rounded = float(np.round(error, 3))
 
     # Markdown display
-    features_md = "### Input Features\n\n" + row.to_frame().to_markdown()
     result_md = (
-        f"### Results\n\n"
+        f"### Prediction Result\n\n"
         f"**Predicted Yield (g per pot):** {pred_rounded}\n\n"
         f"**Actual Yield (Ground Truth):** {actual}\n\n"
         f"**Absolute Error:** {error_rounded}"
     )
 
-    return "✅ Prediction complete", features_md, result_md
+    return "✅ Prediction complete", result_md
 
 # -------------------------------
 # Dropdown options
@@ -79,7 +78,7 @@ body {background:#f5f7fa;color:#0b1220;font-family:Inter,sans-serif;}
 """) as demo:
     
     with gr.Row():
-        # LEFT COLUMN: controls
+        # LEFT COLUMN: controls + selected row info
         with gr.Column(elem_classes="card", scale=1):
             gr.Markdown("# 🌿 EcoGrowAI — Yield Prediction")
             gr.Markdown("Select any experimental row and predict **Yield (g per pot)** using the trained XGBoost model.")
@@ -90,18 +89,17 @@ body {background:#f5f7fa;color:#0b1220;font-family:Inter,sans-serif;}
             
             predict_button = gr.Button("Predict", variant="primary")
             status = gr.Markdown("")
+            row_info = gr.Markdown("No row selected yet.")  # Selected row info on left
 
-        # RIGHT COLUMN: display features and results
-        with gr.Column(elem_classes="card", scale=2):
-            row_info = gr.Markdown("No row selected yet.")
-            features_md = gr.Markdown("")
-            result_md = gr.Markdown("")
+        # RIGHT COLUMN: prediction result
+        with gr.Column(elem_classes="card", scale=1):
+            result_md = gr.Markdown("Prediction result will appear here.")
 
     # -------------------------------
     # Event bindings
     # -------------------------------
     row_dropdown.change(on_row_select, row_dropdown, row_info)
-    predict_button.click(on_predict, row_dropdown, [status, features_md, result_md])
+    predict_button.click(on_predict, row_dropdown, [status, result_md])
 
 # -------------------------------
 # Launch