Update app.py

app.py

@@ -2,6 +2,7 @@ import streamlit as st
 import pandas as pd
 import json
 import io
+import math
 
 # Function to load the .ifs file
 def load_ifs_file(file):
@@ -17,7 +18,6 @@ def extract_values(json_data, code_neo_list):
         try:
             # Navigate to "data -> modules -> food_8 -> questions"
             value = json_data.get("data", {}).get("modules", {}).get("food_8", {}).get("questions", {})
-            # Extract keys after the fixed prefix
             keys = code.replace("data_modules_food_8_questions_", "").split("_")
             for key in keys:
                 if isinstance(value, list):  # Handle lists
@@ -32,10 +32,14 @@ def extract_values(json_data, code_neo_list):
             extracted_values.append(None)
     return extracted_values
 
-# Function to update JSON with new values
-def update_json(json_data, code_neo_list, new_values):
+# Function to safely update JSON with new values while preserving original data
+def update_json_safe(json_data, code_neo_list, new_values):
     for code, new_value in zip(code_neo_list, new_values):
         try:
+            # Skip if new_value is NaN or None to preserve original value
+            if new_value is None or (isinstance(new_value, float) and math.isnan(new_value)):
+                continue  
+            
             # Navigate to the "data -> modules -> food_8 -> questions" level
             value = json_data.get("data", {}).get("modules", {}).get("food_8", {}).get("questions", {})
             keys = code.replace("data_modules_food_8_questions_", "").split("_")
@@ -114,7 +118,7 @@ elif mode == "Update Mode":
             updated_values = updated_df["TEXTE_NEO"].tolist()
             
             # Update JSON
-            updated_json_data = update_json(json_data, code_neo_list, updated_values)
+            updated_json_data = update_json_safe(json_data, code_neo_list, updated_values)
             
             # Provide updated JSON for download
             updated_json_str = json.dumps(updated_json_data, indent=4)