Update app.py

app.py

@@ -4,6 +4,8 @@ import pandas as pd
 import numpy as np
 from scipy.stats import multivariate_normal as mvn
 import random
+import json
+import requests
 # import plotly.graph_objects as go
 
 
@@ -69,6 +71,12 @@ def get_density(city):
 
 
 def get_energy_yield(production, injection, group, length=30):
+    
+    if group == "" or injection is None or production is None:
+        total_text = "Make sure to select injection well, production well, and enter your group number"
+        plt = gr.ScatterPlot()
+        return total_text, plt
+    
     density = get_density(production)
     gamma = gammas[cities.index(production)]
     dist = random.uniform(0.98, 1.02)
@@ -91,13 +99,10 @@ def get_energy_yield(production, injection, group, length=30):
     total_text = f"Total Energy Yield for 30 years: {total_yield:.2f} MWh"
     
     #if group is empty, return warning as text label
-    if group == "":
-        total_text = "Please enter your group number"
-        plt = gr.ScatterPlot()
-        return total_text, plt
-    else:
-        log_final_decision(group, total_yield, injection, production, SUMMARY_TABLE_URL)
-        return total_text, plt
+    
+
+    log_final_decision(group, total_yield, injection, production, SUMMARY_TABLE_URL)
+    return total_text, plt