Update app.py

app.py

@@ -266,13 +266,10 @@ def ask_judge(question, conversation, current_topic):
         messages=[{"role": "user", "content": f"{context}\nSpecific Question: {question}"}],
         temperature=0.6
     )
-    embed_and_store(response, "Judge")
-    return response
-
+    
 def step(topic_input, conversation, turn_count, current_topic, last_ruling_turn, agent_params):
     """Advance the discussion by one turn"""
-    global current_topic, auto_mode
-    
+    # Remove global declarations - we'll use the parameters directly
     # Set topic on first turn
     if turn_count == 0:
         if topic_input.strip():
@@ -326,7 +323,7 @@ def step(topic_input, conversation, turn_count, current_topic, last_ruling_turn,
     new_last_ruling_turn = new_turn_count if agent_name == "Judge" else last_ruling_turn
     
     # Update memory
-    embed_and_store(response, agent_name)
+    embed_and_store(response, agent_name, current_topic)  # Pass current_topic here
     
     # Format HTML output
     html_output = format_conversation_html(updated_conversation)
@@ -359,6 +356,26 @@ def step(topic_input, conversation, turn_count, current_topic, last_ruling_turn,
         agent_params
     )
 
+# Update embed_and_store to accept topic as parameter
+def embed_and_store(text, agent=None, topic=""):
+    """Store text with associated topic"""
+    try:
+        vec = get_embedding(text)
+        memory_index.add(np.array([vec], dtype='float32'))
+        memory_data.append({
+            "text": text,
+            "timestamp": datetime.now().isoformat(),
+            "agent": agent or "system",
+            "topic": topic
+        })
+        if len(memory_data) % 5 == 0:
+            with open(MEMORY_FILE, "wb") as f:
+                pickle.dump(memory_data, f)
+            faiss.write_index(memory_index, INDEX_FILE)
+    except Exception as e:
+        print(f"Memory Error: {str(e)}")
+
+# ... [Rest of the functions remain unchanged] ...
 def get_last_by_agent(conversation, agent_name):
     """Get last message from specific agent"""
     for msg in reversed(conversation):