Update app.py

app.py

@@ -1,155 +1,155 @@
-from flask import Flask, render_template, request, redirect, url_for, send_file, flash, jsonify
-import os
-import traceback
-from pathlib import Path
-import json
-import pandas as pd
-
-app = Flask(__name__)
-app.secret_key = os.urandom(24)
-
-try:
-    import main as project_main
-except Exception as e:
-    project_main = None
-    import_error = traceback.format_exc()
-else:
-    import_error = None
-
-RESULTS_CSV = Path("results_flask.csv")
-
-def safe_update_dict_from_json(orig_dict, json_text):
-    if not json_text or not json_text.strip():
-        return orig_dict
-    try:
-        new = json.loads(json_text)
-        if not isinstance(new, dict):
-            return orig_dict
-        d = orig_dict.copy()
-        d.update(new)
-        return d
-    except Exception:
-        return orig_dict
-
-@app.route("/", methods=["GET"])
-def index():
-    if project_main is None:
-        return render_template("index.html", import_error=import_error, project=None)
-    techs = list(project_main.TECHNOLOGY_DATA.keys())
-    defaults = {
-        "inflation": project_main.INFLATION_RATE,
-        "tax": project_main.TAX_RATE,
-        "years": project_main.PROJECT_YEARS,
-        "cap_min": project_main.OPTIMIZATION_SPACE['capacity_kta'][0],
-        "cap_max": project_main.OPTIMIZATION_SPACE['capacity_kta'][1],
-        "export_mix": project_main.OPTIMIZATION_SPACE['export_market_mix'][0],
-    }
-    return render_template("index.html", import_error=None, project=project_main, techs=techs, defaults=defaults)
-
-@app.route("/run", methods=["POST"])
-def run():
-    if project_main is None:
-        flash("خطا: فایل main.py بارگزاری نشد. لطفاً لاگ را بررسی کنید.", "danger")
-        return redirect(url_for("index"))
-    try:
-        inflation_rate = float(request.form.get("inflation", project_main.INFLATION_RATE))
-        tax_rate = float(request.form.get("tax", project_main.TAX_RATE))
-        project_years = int(request.form.get("years", project_main.PROJECT_YEARS))
-        capacity_min = float(request.form.get("cap_min", project_main.OPTIMIZATION_SPACE['capacity_kta'][0]))
-        capacity_max = float(request.form.get("cap_max", project_main.OPTIMIZATION_SPACE['capacity_kta'][1]))
-        technology = request.form.get("technology", list(project_main.TECHNOLOGY_DATA.keys())[0])
-        export_mix = float(request.form.get("export_mix", project_main.OPTIMIZATION_SPACE['export_market_mix'][0]))
-        sell_byproducts = request.form.get("sell_byproducts") == "on"
-        tech_json = request.form.get("tech_json", "")
-        prices_json = request.form.get("prices_json", "")
-    except Exception as e:
-        flash("خطا در خواندن ورودی‌ها: " + str(e), "danger")
-        return redirect(url_for("index"))
-
-    try:
-        project_main.INFLATION_RATE = inflation_rate
-        project_main.TAX_RATE = tax_rate
-        project_main.PROJECT_YEARS = project_years
-
-        project_main.OPTIMIZATION_SPACE['capacity_kta'] = (capacity_min, capacity_max)
-        if technology not in project_main.OPTIMIZATION_SPACE.get('technology', []):
-            project_main.OPTIMIZATION_SPACE['technology'] = [technology]
-        low = max(0.0, export_mix - 0.05)
-        high = min(1.0, export_mix + 0.05)
-        if low == high:
-            high = min(1.0, high + 0.01)
-        project_main.OPTIMIZATION_SPACE['export_market_mix'] = (low, high)
-        project_main.OPTIMIZATION_SPACE['sell_byproducts'] = [bool(sell_byproducts)]
-
-        project_main.TECHNOLOGY_DATA = safe_update_dict_from_json(project_main.TECHNOLOGY_DATA, tech_json)
-        project_main.PRODUCT_PRICES_USD_PER_TON = safe_update_dict_from_json(project_main.PRODUCT_PRICES_USD_PER_TON, prices_json)
-    except Exception as e:
-        flash("خطا در اعمال پارامترها: " + str(e), "danger")
-        return redirect(url_for("index"))
-
-    try:
-        flash("محاسبات شروع شد — صبر کنید تا عملیات به پایان برسد...", "info")
-        results = project_main.run_optimizations_without_ml()
-        df_results = pd.DataFrame(results).sort_values(by="irr", ascending=False).reset_index(drop=True)
-        df_results = df_results.round(2)
-
-        df_results.to_csv(RESULTS_CSV, index=False, encoding='utf-8-sig')
-
-        try:
-            project_main.display_and_save_results(df_results)
-            project_main.create_kpi_comparison_dashboard(df_results)
-        except Exception:
-            pass
-
-        if not df_results.empty:
-            best = df_results.iloc[0]
-            top_kpis = {
-                "irr": round(float(best['irr']), 2),
-                "annual_profit_M": round(float(best['annual_profit'])/1_000_000, 2),
-                "capex_M": round(float(best['total_capex'])/1_000_000, 2),
-                "payback": round(float(best['payback_period']), 2)
-            }
-        else:
-            top_kpis = None
-
-        charts_data = df_results.to_dict(orient="records")
-
-        flash("محاسبات با موفقیت تکمیل شد.", "success")
-     
-        kpi_img = Path("static/images/kpi_dashboard.png") if Path("static/images/kpi_dashboard.png").exists() else None
-        tornado_img = Path("static/images/sensitivity_analysis_tornado.png") if Path("static/images/sensitivity_analysis_tornado.png").exists() else None
-
-        return render_template(
-            "index.html",
-            project=project_main,
-            techs=list(project_main.TECHNOLOGY_DATA.keys()),
-            defaults={"inflation": project_main.INFLATION_RATE, "tax": project_main.TAX_RATE},
-            table_html = df_results.to_html(
-                classes='table table-striped table-dark',
-                index=False,
-                justify='center',
-                float_format=lambda x: f"{x:.2f}"
-                ),
-            top_kpis=top_kpis,
-            charts_json=json.dumps(charts_data, default=str),
-            kpi_img="images/kpi_dashboard.png" if kpi_img else None,
-            tornado_img="images/sensitivity_analysis_tornado.png" if tornado_img else None
-        )
-
-    except Exception as e:
-        tb = traceback.format_exc()
-        flash("خطا در اجرای الگوریتم‌ها. لاگ: " + str(e), "danger")
-        return render_template("index.html", import_error=tb, project=project_main)
-
-@app.route("/download")
-def download_results():
-    if RESULTS_CSV.exists():
-        return send_file(str(RESULTS_CSV), as_attachment=True)
-    flash("فایل نتایج آماده نیست.", "warning")
-    return redirect(url_for("index"))
-
-if __name__ == "__main__":
-    app.run(debug=True)
-    
-    
+from flask import Flask, render_template, request, redirect, url_for, send_file, flash, jsonify
+import os
+import traceback
+from pathlib import Path
+import json
+import pandas as pd
+
+app = Flask(__name__)
+app.secret_key = os.urandom(24)
+
+try:
+    import main as project_main
+except Exception as e:
+    project_main = None
+    import_error = traceback.format_exc()
+else:
+    import_error = None
+
+RESULTS_CSV = Path("results_flask.csv")
+
+def safe_update_dict_from_json(orig_dict, json_text):
+    if not json_text or not json_text.strip():
+        return orig_dict
+    try:
+        new = json.loads(json_text)
+        if not isinstance(new, dict):
+            return orig_dict
+        d = orig_dict.copy()
+        d.update(new)
+        return d
+    except Exception:
+        return orig_dict
+
+@app.route("/", methods=["GET"])
+def index():
+    if project_main is None:
+        return render_template("index.html", import_error=import_error, project=None)
+    techs = list(project_main.TECHNOLOGY_DATA.keys())
+    defaults = {
+        "inflation": project_main.INFLATION_RATE,
+        "tax": project_main.TAX_RATE,
+        "years": project_main.PROJECT_YEARS,
+        "cap_min": project_main.OPTIMIZATION_SPACE['capacity_kta'][0],
+        "cap_max": project_main.OPTIMIZATION_SPACE['capacity_kta'][1],
+        "export_mix": project_main.OPTIMIZATION_SPACE['export_market_mix'][0],
+    }
+    return render_template("index.html", import_error=None, project=project_main, techs=techs, defaults=defaults)
+
+@app.route("/run", methods=["POST"])
+def run():
+    if project_main is None:
+        flash("خطا: فایل main.py بارگزاری نشد. لطفاً لاگ را بررسی کنید.", "danger")
+        return redirect(url_for("index"))
+    try:
+        inflation_rate = float(request.form.get("inflation", project_main.INFLATION_RATE))
+        tax_rate = float(request.form.get("tax", project_main.TAX_RATE))
+        project_years = int(request.form.get("years", project_main.PROJECT_YEARS))
+        capacity_min = float(request.form.get("cap_min", project_main.OPTIMIZATION_SPACE['capacity_kta'][0]))
+        capacity_max = float(request.form.get("cap_max", project_main.OPTIMIZATION_SPACE['capacity_kta'][1]))
+        technology = request.form.get("technology", list(project_main.TECHNOLOGY_DATA.keys())[0])
+        export_mix = float(request.form.get("export_mix", project_main.OPTIMIZATION_SPACE['export_market_mix'][0]))
+        sell_byproducts = request.form.get("sell_byproducts") == "on"
+        tech_json = request.form.get("tech_json", "")
+        prices_json = request.form.get("prices_json", "")
+    except Exception as e:
+        flash("خطا در خواندن ورودی‌ها: " + str(e), "danger")
+        return redirect(url_for("index"))
+
+    try:
+        project_main.INFLATION_RATE = inflation_rate
+        project_main.TAX_RATE = tax_rate
+        project_main.PROJECT_YEARS = project_years
+
+        project_main.OPTIMIZATION_SPACE['capacity_kta'] = (capacity_min, capacity_max)
+        if technology not in project_main.OPTIMIZATION_SPACE.get('technology', []):
+            project_main.OPTIMIZATION_SPACE['technology'] = [technology]
+        low = max(0.0, export_mix - 0.05)
+        high = min(1.0, export_mix + 0.05)
+        if low == high:
+            high = min(1.0, high + 0.01)
+        project_main.OPTIMIZATION_SPACE['export_market_mix'] = (low, high)
+        project_main.OPTIMIZATION_SPACE['sell_byproducts'] = [bool(sell_byproducts)]
+
+        project_main.TECHNOLOGY_DATA = safe_update_dict_from_json(project_main.TECHNOLOGY_DATA, tech_json)
+        project_main.PRODUCT_PRICES_USD_PER_TON = safe_update_dict_from_json(project_main.PRODUCT_PRICES_USD_PER_TON, prices_json)
+    except Exception as e:
+        flash("خطا در اعمال پارامترها: " + str(e), "danger")
+        return redirect(url_for("index"))
+
+    try:
+        flash("محاسبات شروع شد — صبر کنید تا عملیات به پایان برسد...", "info")
+        results = project_main.run_optimizations_without_ml()
+        df_results = pd.DataFrame(results).sort_values(by="irr", ascending=False).reset_index(drop=True)
+        df_results = df_results.round(2)
+
+        df_results.to_csv(RESULTS_CSV, index=False, encoding='utf-8-sig')
+
+        try:
+            project_main.display_and_save_results(df_results)
+            project_main.create_kpi_comparison_dashboard(df_results)
+        except Exception:
+            pass
+
+        if not df_results.empty:
+            best = df_results.iloc[0]
+            top_kpis = {
+                "irr": round(float(best['irr']), 2),
+                "annual_profit_M": round(float(best['annual_profit'])/1_000_000, 2),
+                "capex_M": round(float(best['total_capex'])/1_000_000, 2),
+                "payback": round(float(best['payback_period']), 2)
+            }
+        else:
+            top_kpis = None
+
+        charts_data = df_results.to_dict(orient="records")
+
+        flash("محاسبات با موفقیت تکمیل شد.", "success")
+     
+        kpi_img = Path("static/images/kpi_dashboard.png") if Path("static/images/kpi_dashboard.png").exists() else None
+        tornado_img = Path("static/images/sensitivity_analysis_tornado.png") if Path("static/images/sensitivity_analysis_tornado.png").exists() else None
+
+        return render_template(
+            "index.html",
+            project=project_main,
+            techs=list(project_main.TECHNOLOGY_DATA.keys()),
+            defaults={"inflation": project_main.INFLATION_RATE, "tax": project_main.TAX_RATE},
+            table_html = df_results.to_html(
+                classes='table table-striped table-dark',
+                index=False,
+                justify='center',
+                float_format=lambda x: f"{x:.2f}"
+                ),
+            top_kpis=top_kpis,
+            charts_json=json.dumps(charts_data, default=str),
+            kpi_img="images/kpi_dashboard.png" if kpi_img else None,
+            tornado_img="images/sensitivity_analysis_tornado.png" if tornado_img else None
+        )
+
+    except Exception as e:
+        tb = traceback.format_exc()
+        flash("خطا در اجرای الگوریتم‌ها. لاگ: " + str(e), "danger")
+        return render_template("index.html", import_error=tb, project=project_main)
+
+@app.route("/download")
+def download_results():
+    if RESULTS_CSV.exists():
+        return send_file(str(RESULTS_CSV), as_attachment=True)
+    flash("فایل نتایج آماده نیست.", "warning")
+    return redirect(url_for("index"))
+
+if __name__ == "__main__":
+    app.run(host="0.0.0.0", port=int(os.environ.get("PORT", 7860)))
+    
+