Update app.py

app.py

@@ -3,18 +3,18 @@ import json
 import google.generativeai as genai
 
 
-def add_to_json(goal):  # Renamed 'new_goal' to 'goal' for clarity
-    with open("test.json", "r") as file:
-        data = json.load(file)
+def add_to_json(goal):
+    try:
+        with open("test.json", "r") as file:
+            data = json.load(file)
+    except FileNotFoundError:
+        data = {"goals": []}  # Create an empty "goals" list if the file doesn't exist
 
-    new_item = {
-        "Goal": goal,
-        # "reminder": new_reminder  # Remove if not used
-    }
-    # data["goal"].append(new_item)
+    new_item = {"Goal": goal}
+    data["goals"].append(new_item)
 
-    # with open("test.json", "w") as file:
-    #     json.dump(data, file, indent=4)
+    with open("test.json", "w") as file:
+        json.dump(data, file, indent=4)
 
 
 GOOGLE_API_KEY = "AIzaSyCUBaL7TdISL7lRuBy19_X0-OsZfgbIgEc"
@@ -24,7 +24,7 @@ model = genai.GenerativeModel('gemini-pro')
 
 def main():  # Wrap main logic in a function
     if prompt := st.chat_input("Hi, how can I help you?"):
-        goal = prompt  # Capture the user's goal
+        goals = prompt  # Capture the user's goal
 
         goals_prompt = f"""Act as a personal assistant. Understand user intent from the point of view for asking a few questions. Ask to know more details about {goal}    :-  {goal}"""  
         completion = model.generate_content(goals_prompt)