app.py

app.py

@@ -47,15 +47,29 @@ if not OPENAI_API_KEY:
 # Tabs
 tab1, tab2 = st.tabs(["Contract", "Technical"])
 
-# Function to save flagged responses
+FLAGGED_RESPONSES_FILE = "/mnt/data/flagged_responses.json"
+
 def save_flagged_response(user_query, ai_response):
     flagged_data = {
         "query": user_query,
         "response": ai_response,
         "timestamp": datetime.now().isoformat()
     }
-    with open("flagged_responses.json", "a") as file:
-        file.write(json.dumps(flagged_data) + "\n")
+    
+    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 = []
+    
+    existing_data.append(flagged_data)
+    
+    with open(FLAGGED_RESPONSES_FILE, "w") as file:
+        json.dump(existing_data, file, indent=4)
+    
     st.success("Response flagged for review.")
 
 # Contract Chat Section
@@ -121,4 +135,10 @@ ASSISTANT_CONTRACT_ID = "asst_rd9h8PfYuOmHbkvOF3RTmVfn"
 ASSISTANT_TECHNICAL_ID = "asst_xizNZBCJuy4TqdjqjwkxbAki"
 
 contract_chat_section(tab1, ASSISTANT_CONTRACT_ID, "contract_messages", "contract_input")
-contract_chat_section(tab2, ASSISTANT_TECHNICAL_ID, "technical_messages", "technical_input")
+contract_chat_section(tab2, ASSISTANT_TECHNICAL_ID, "technical_messages", "technical_input")
+
+# Provide download option for flagged responses
+if os.path.exists(FLAGGED_RESPONSES_FILE):
+    with open(FLAGGED_RESPONSES_FILE, "r") as file:
+        flagged_responses = file.read()
+    st.download_button("Download Flagged Responses", data=flagged_responses, file_name="flagged_responses.json", mime="application/json")