app.py

app.py

@@ -50,35 +50,22 @@ if not OPENAI_API_KEY or not ASSISTANT_CONTRACT_ID or not ASSISTANT_TECHNICAL_ID
 # Tabs
 tab1, tab2, tab3 = st.tabs(["Contract", "Technical", "Flagged Responses"])
 
-FLAGGED_RESPONSES_FILE = "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()
+FLAGGED_RESPONSES_DIR = "flagged_responses"
+os.makedirs(FLAGGED_RESPONSES_DIR, exist_ok=True)
 
 def save_flagged_response(user_query, ai_response):
+    timestamp = datetime.now().strftime("%Y%m%d_%H%M%S")
+    filename = f"{FLAGGED_RESPONSES_DIR}/flagged_{timestamp}.json"
     flagged_data = {
         "query": user_query,
         "response": ai_response,
         "timestamp": datetime.now().isoformat()
     }
     
-    with open(FLAGGED_RESPONSES_FILE, "r") as file:
-        try:
-            existing_data = json.load(file)
-        except json.JSONDecodeError:
-            existing_data = []
-    
-    existing_data.append(flagged_data)
-    
-    with open(FLAGGED_RESPONSES_FILE, "w") as file:
-        json.dump(existing_data, file, indent=4)
+    with open(filename, "w") as file:
+        json.dump(flagged_data, file, indent=4)
     
-    st.success("Response flagged for review.")
+    st.success(f"Response flagged and saved as {filename}.")
 
 # Contract Chat Section
 def contract_chat_section(tab, assistant_id, session_key, input_key):
@@ -144,11 +131,14 @@ contract_chat_section(tab2, ASSISTANT_TECHNICAL_ID, "technical_messages", "techn
 # Flagged Responses Tab
 with tab3:
     st.subheader("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")
+    flagged_files = [f for f in os.listdir(FLAGGED_RESPONSES_DIR) if f.endswith(".json")]
+    
+    if flagged_files:
+        selected_file = st.selectbox("Select a flagged response file to download:", flagged_files)
+        
+        if selected_file:
+            with open(os.path.join(FLAGGED_RESPONSES_DIR, selected_file), "r") as file:
+                flagged_responses = file.read()
+            st.download_button("Download Selected Flagged Responses", data=flagged_responses, file_name=selected_file, mime="application/json")
     else:
-        st.info("No flagged responses available.")
-
-
+        st.info("No flagged responses available.")