Update app.py

app.py

@@ -14,15 +14,37 @@ CITATION_BUTTON_TEXT = """@article{2022,
    month={Jun} }
 """               
 
-# Load the JSON data
+def initialize_healthsheet(json_data):
+    healthsheet_dict = {}
+    for header in json_data:
+        healthsheet_dict[header] = {}
+        for var_name, question in json_data[header].items():
+            healthsheet_dict[header][var_name] = {
+                'question': question,
+                'answer': None
+            }
+    return healthsheet_dict
+
+def print_healthsheet(healthsheet_dict):
+    healthsheet = ""
+    for header in healthsheet_dict:
+        healthsheet += f"# {header}\n\n"
+        for var_name in json_data[header]:
+            healthsheet += f"## {var_name['question']}\n\n"
+            healthsheet += f"## {var_name['answer']}\n\n"            
+    return healthsheet
+
+def update_healthsheet(healthsheet_dict, header, var_name, answer):
+    healthsheet_dict[header][var_name] = answer
+    healthsheet = print_healthsheet(healthsheet_dict)
+    return healthsheet_dict, healthsheet
+
+
 with open('healthsheet.json', 'r') as file:
     json_data = json.load(file)
 
-healthsheet = ""
-for header in json_data:
-    healthsheet += f"# {header}\n\n"
-    for var_name, question in json_data[header].items():
-       healthsheet += f"## {question}\n\n"
+healthsheet_dict = initialize_healthsheet(json_data)
+healthsheet = print_healthsheet(healthsheet_dict)
 
 with gr.Blocks() as demo:
     gr.HTML("<h1><center>Healthsheet Creator! 🪄📄✨<h1><center>")
@@ -44,8 +66,9 @@ with gr.Blocks() as demo:
             for tab_name in json_data:
                 with gr.Tab(tab_name):
                     for var_name, label_text in json_data[tab_name].items():
-                        gr.Textbox(name=var_name, label=label_text, lines=3)
-                    gr.Button("Enter")
+                        answer = gr.Textbox(label=label_text, lines=3)
+                        update_btn = gr.Button("Update")
+                        update_btn.click(update_healthsheet, input=[healthsheet_dict, header, var_name, answer], output=[healthsheet_dict, healthsheet])
         
         with gr.Column():
             gr.Markdown("This is where your Markdown-formatted healthsheet will appear! You can copy and paste it into your dataset card.")