app.py

app.py

@@ -49,6 +49,14 @@ tab1, tab2 = st.tabs(["Contract", "Technical"])
 
 FLAGGED_RESPONSES_FILE = "/mnt/data/flagged_responses.json"
 
+# Ensure the flagged responses file exists
+def ensure_flagged_file():
+    if not os.path.exists(FLAGGED_RESPONSES_FILE):
+        with open(FLAGGED_RESPONSES_FILE, "w") as file:
+            json.dump([], file)  # Initialize with an empty list
+
+ensure_flagged_file()
+
 def save_flagged_response(user_query, ai_response):
     flagged_data = {
         "query": user_query,
@@ -56,14 +64,11 @@ def save_flagged_response(user_query, ai_response):
         "timestamp": datetime.now().isoformat()
     }
     
-    if os.path.exists(FLAGGED_RESPONSES_FILE):
-        with open(FLAGGED_RESPONSES_FILE, "r") as file:
-            try:
-                existing_data = json.load(file)
-            except json.JSONDecodeError:
-                existing_data = []
-    else:
-        existing_data = []
+    with open(FLAGGED_RESPONSES_FILE, "r") as file:
+        try:
+            existing_data = json.load(file)
+        except json.JSONDecodeError:
+            existing_data = []
     
     existing_data.append(flagged_data)