add graph

app.py

@@ -85,11 +85,14 @@ def optimize_dynamic(
     result_text += f"\n\n⚡️ Predicted Fuel Rate: {result.fun:.2f} L/cycle"
 
     # Sensitivity analysis for Engine Speed
-    sensitivity_lines = ["\n📊 Suggested RPM vs. Fuel Rate:"]
+    # sensitivity_lines = ["\n📊 Suggested RPM vs. Fuel Rate:"]
     start_rpm = int(round(final_input['Eng_Speed_Ave'] / 50.0) * 50)
     end_rpm = min(start_rpm + 300, bounds_dict['Eng_Speed_Ave'][1])
     start_rpm = max(start_rpm, bounds_dict['Eng_Speed_Ave'][0])
 
+    rpm_values = []
+    fuel_rates = []
+
     for rpm in range(start_rpm, end_rpm + 1, 50):
         temp_input = final_input.copy()
         temp_input['Eng_Speed_Ave'] = rpm
@@ -114,10 +117,23 @@ def optimize_dynamic(
         else:
             fuel_rate = best_model.predict(pd.DataFrame([temp_input]))[0]
 
-        sensitivity_lines.append(f"RPM {rpm}: {fuel_rate:.2f} L/cycle")
+        rpm_values.append(rpm)
+        fuel_rates.append(fuel_rate)
+
+    # Create plot
+    plt.figure(figsize=(6, 4))
+    plt.plot(rpm_values, fuel_rates, marker='o')
+    plt.title("Sensitivity of Fuel Rate to Engine Speed")
+    plt.xlabel("Engine Speed (RPM)")
+    plt.ylabel("Predicted Fuel Rate (L/cycle)")
+    plt.grid(True)
 
-    result_text += "\n" + "\n".join(sensitivity_lines)
-    return result_text
+    # Save and return plot image
+    plot_path = "sensitivity_plot.png"
+    plt.savefig(plot_path)
+    plt.close()
+
+    return result_text, plot_path
 
 # Gradio Interface (Interface-style)
 interface = gr.Interface(
@@ -136,7 +152,10 @@ interface = gr.Interface(
         gr.Checkbox(label="Fix Loading Stop"),
         gr.Checkbox(label="Fix Loaded Stop"),
     ],
-    outputs=gr.Textbox(label="Optimization Result", lines=16),
+        outputs=[
+        gr.Textbox(label="Optimization Result"),
+        gr.Image(type="filepath", label="Sensitivity Plot")
+    ],
     title="⚙️ Fuel Rate What-If Optimizer",
     description="Perform global optimization of fuel rate with optional fixed inputs and sensitivity on engine RPM."
 )