Update app.py

app.py

@@ -12,7 +12,9 @@ model.eval()
 
 # Define the prediction function
 def predict_water_usage(state_idx, target_year, structured_data):
-    # Ensure structured_data is correctly formatted
+    # Convert structured data JSON string to dictionary
+    structured_data = eval(structured_data) if isinstance(structured_data, str) else structured_data
+
     if state_idx not in structured_data or len(structured_data[state_idx]) < 5:
         return {"error": "Structured data must include 5 years of data for the specified state."}
 
@@ -26,14 +28,14 @@ def predict_water_usage(state_idx, target_year, structured_data):
     
     return {"prediction": output.tolist()}
 
-# Configure Gradio interface with structured JSON input example
+# Configure Gradio interface
 inputs = [
     gr.components.Number(label="State Index"),
     gr.components.Number(label="Target Year"),
-    gr.components.JSON(
-        label="Structured Data", 
-        placeholder="""
-{
+    gr.components.Textbox(
+        label="Structured Data (JSON format)",
+        lines=10,
+        placeholder="""{
   "state_idx": {
     "2020": [value1, value2, ..., value8],
     "2021": [value1, value2, ..., value8],
@@ -41,8 +43,7 @@ inputs = [
     "2023": [value1, value2, ..., value8],
     "2024": [value1, value2, ..., value8]
   }
-}
-"""
+}"""
     )
 ]
 outputs = gr.components.JSON(label="Prediction")