Add effect of temp change on trees

app.py

@@ -2,41 +2,15 @@
 
 import os
 from shiny import reactive, render, ui, App
-# from shiny.express import input, render, ui
-from shinywidgets import render_widget
 
 from plot_utils import show_one_plot, show_two_plots
-from data_utils import prepare_data, process_data_trees_to_temp
+from data_utils import prepare_data, process_data_trees_to_temp, process_data_temp_to_trees
 
 import pandas as pd
 import geopandas as gpd
 
-# import matplotlib.pyplot as plt
-# import matplotlib.colors as colors
-# import matplotlib.cm as cm
-# import matplotlib.gridspec as gridspec
 import numpy as np
 
-import requests
-
-url = "https://Projects-by-IF-model-temp-api.hf.space/model-effect"
-headers = {"Content-Type": "application/json"}
-
-####test####
-data = {
-    "X": [[1462, -0.18, 234, 8]], "T0":[10], "T1":[15]
-}
-
-response = requests.post(url, headers=headers, json=data)
-
-if response.ok:
-    result = response.json()
-    print("Model response:", result["effect"])
-else:
-    print("Request failed:", response.status_code, response.text)
-#######
-
-# treatment_df = pd.read_csv("example.csv", index_col=0)
 og_file = "cleaned_dataframe.csv"
 training_file = "training_dataframe.csv"
 original_df, df, X, Y, T = prepare_data(og_file, training_file)
@@ -57,14 +31,20 @@ def server(input, output, session):
         if input.tab_choice() == "Effect of trees on temperature":
             return ui.input_slider("tree_pct", "Tree Increase (%)", min=0, max=100, value=40, step=5)
         else:  # "Effect of temperature on trees"
-            return ui.input_slider("temp_goal", "Temperature Decrease (°C)", min=0, max=5, value=1, step=0.2)
+            return ui.input_slider("temp_goal", "Temperature Decrease (°C)", min=0, max=2, value=0.5, step=0.1)
             
  
     @reactive.Calc
     def get_gdf_trees():
         value_trees = input.tree_pct()
-        print(f"value inputed is {input.tree_pct()}")
+        print(f"value inputed is {value_trees}")
         return process_data_trees_to_temp(value_trees, original_df, df, X, Y, T)
+
+    @reactive.Calc
+    def get_gdf_temp():
+        value_temp = input.temp_goal()
+        print(f"value inputed is {value_temp}")
+        return process_data_temp_to_trees(value_temp, original_df, df, X, Y, T)
     
     @output
     @render.ui
@@ -73,12 +53,13 @@ def server(input, output, session):
             return ui.div(
                 ui.h2("Temperature Simulation Viewer", class_="text-center"),
                 ui.div(ui.output_plot("temp_after_treatment"), style="display: flex; justify-content: center;"),
-                ui.div(ui.output_plot("treatment_trees_effect"), style="display: flex; justify-content: center;"),
+                ui.div(ui.output_plot("temp_change"), style="display: flex; justify-content: center;"),
             )
         else: #Effect of temperature on trees
             return ui.div(
                 ui.h2("Tree Simulation Viewer", class_="text-center"),
-                ui.div(ui.output_plot("temp_after_treatment"), style="display: flex; justify-content: center;"),
+                ui.div(ui.output_plot("trees_after_decrease"), style="display: flex; justify-content: center;"),
+                ui.div(ui.output_plot("tree_coverage_change"), style="display: flex; justify-content: center;"),
             )
 
     @output
@@ -100,15 +81,44 @@ def server(input, output, session):
 
     @output
     @render.plot()
-    def treatment_trees_effect():
+    def temp_change():
         gdf_trees = get_gdf_trees()
         value_trees = input.tree_pct()
-        print(value_trees)
-        print(gdf_trees)
         fig = show_one_plot(gdf_trees, f'treatment_effect_{value_trees}%', -2, 0, 'PuBu_r', 
                   f"Decrease of temperature from treatment (+{value_trees}% Trees)", 
                   "Temperature decrease (°C)")
         return fig
     
+    @output
+    @render.plot()
+    def trees_after_decrease():
+        gdf_temp = get_gdf_temp()
+        value_temp = input.temp_goal()
+        min_cobertura_veg = 5
+        max_cobertura = 100
+        try:
+            fig = show_two_plots(gdf_temp, f'total_trees_needed_for_{value_temp}C', '%CoberturaVeg', 
+             min_cobertura_veg, max_cobertura, 'YlGn', 
+             f"Tree coverage needed for {value_temp}°C decrease","Original tree canopy coverage",
+             "Tree canopy coverage (%)", split=True)
+            return fig
+        except Exception as e:
+            print("Plotting error:", e)
+            raise e
+
+
+    @output
+    @render.plot()
+    def tree_coverage_change():
+        gdf_temp = get_gdf_temp()
+        value_temp = input.temp_goal()
+        min_tree_increase = 0
+        max_tree_increase = 90
+        fig = show_one_plot(gdf_temp, f'pp_trees_increase_for_{value_temp}C', min_tree_increase, 
+                  max_tree_increase, 'Greens', 
+                  f"Increase in tree coverage for {value_temp}°C decrease",
+                  "Tree canopy coverage increase in %pt")
+        return fig
+
+app = App(app_ui, server)
 
-app = App(app_ui, server)

data_utils.py

@@ -67,4 +67,34 @@ def process_data_trees_to_temp(value, original_df, df, X, Y, T):
 
     gdf = gpd.GeoDataFrame(treatment_df, geometry=gpd.points_from_xy(treatment_df.x, treatment_df.y))
 
-    return gdf
+    return gdf
+
+
+def run_temp_decrease(X, original_df, df, goal, name):
+    
+    treatment_df = prepare_treatment_df(original_df, df)
+    data = {"X": X.tolist()}
+    response = requests.post(url, headers=headers, json=data)
+    if response.ok:
+        result = response.json()
+        # print("Model response:", result["effect"])
+    else:
+        print("Request failed:", response.status_code, response.text)
+    
+    treatment_effects = np.array(result["effect"])
+    
+    trees_needed = -goal / treatment_effects
+    
+    treatment_df[name] = trees_needed
+
+    return treatment_df
+def process_data_temp_to_trees(goal, original_df, df, X, Y, T):
+
+    name = f'pp_trees_increase_for_{goal}C'
+    treatment_df = run_temp_decrease(X,original_df, df, goal, name)
+
+    treatment_df[f'total_trees_needed_for_{goal}C'] = treatment_df['%CoberturaVeg'] + treatment_df[name]
+
+    gdf = gpd.GeoDataFrame(treatment_df, geometry=gpd.points_from_xy(treatment_df.x, treatment_df.y))
+
+    return gdf

plot_utils.py

@@ -77,19 +77,17 @@ def show_one_plot(gdf, name, vmin, vmax, colorscheme, title, label):
     norm = colors.Normalize(vmin, vmax)
 
     # Create subplots
-    # fig, axes = plt.subplots(1, 1, figsize=(6, 6))
     fig = plt.figure(figsize=(12, 6))
     gs = gridspec.GridSpec(1, 2, width_ratios=[1, 0.05], wspace=0.3)
 
     axes = [fig.add_subplot(gs[0]), fig.add_subplot(gs[1])]
     
-    print(gdf[name].min())
-    # First plot (simulated)
+
     gdf.plot(column=name, cmap=cmap, norm=norm, ax=axes[0], marker='s', markersize=10)
     axes[0].set_title(title)
     axes[0].set_axis_off()
 
-        # Add a shared colorbar
+    # Add a colorbar
     sm = cm.ScalarMappable(cmap=cmap, norm=norm)
     sm._A = []  # Dummy data for the colorbar
     cbar = fig.colorbar(sm, cax=axes[1], orientation='vertical', fraction=0.03, pad=0.02)