Update app.py

app.py

@@ -1,5 +1,5 @@
 import gradio as gr
-from inference import evo_chat_predict, get_model_config, get_system_stats, get_gpt_response, retrain_from_feedback
+from inference import evo_chat_predict, get_model_config, get_gpt_response, retrain_from_feedback
 import pandas as pd
 import csv
 from datetime import datetime
@@ -37,21 +37,22 @@ with gr.Blocks(theme=gr.themes.Base(), css="body { background-color: #0f0f0f; co
         )
 
     # 🔍 Ask Evo
-    def ask_evo(q, opt1, opt2, hist):
+    def ask_evo(q, opt1, opt2, hist, selected):
         result = evo_chat_predict(hist, q, [opt1, opt2])
         evo_text = f"Answer: {result['answer']} (Confidence: {result['confidence']})\n\nReasoning: {result['reasoning']}"
         gpt_text = get_gpt_response(q)
         stats = get_model_config()
         stats_text = f"Layers: {stats['num_layers']} | Heads: {stats['num_heads']} | FFN: {stats['ffn_dim']} | Memory: {stats['memory_enabled']} | Phase: {stats['phase']} | Accuracy: {stats['accuracy']}"
-        
-        # ✅ Safe conversation history update
+
+        # Update history
         new_row = [q, opt1, opt2, result["answer"], result["confidence"], result["reasoning"], result["context_used"]]
         new_row_df = pd.DataFrame([new_row], columns=hist.columns)
-        if hist.empty:
-            updated_df = new_row_df
-        else:
-            updated_df = pd.concat([hist, new_row_df], ignore_index=True)
-        
+        updated_df = new_row_df if hist.empty else pd.concat([hist, new_row_df], ignore_index=True)
+
+        # Log feedback
+        if selected in ["Evo", "GPT"]:
+            feedback_log.append(new_row)
+
         return evo_text, gpt_text, stats_text, updated_df
 
     # 🔁 Actual retraining
@@ -78,7 +79,7 @@ with gr.Blocks(theme=gr.themes.Base(), css="body { background-color: #0f0f0f; co
             return "⚠️ No feedback to export."
 
     # 🔘 Event bindings
-    evo_btn.click(fn=ask_evo, inputs=[query, option1, option2, convo], outputs=[evo_box, gpt_box, evo_stats, convo])
+    evo_btn.click(fn=ask_evo, inputs=[query, option1, option2, convo, feedback], outputs=[evo_box, gpt_box, evo_stats, convo])
     retrain_btn.click(fn=retrain_evo, inputs=[], outputs=[status_box])
     clear_btn.click(fn=clear_fields, inputs=[], outputs=[query, option1, option2, evo_box, gpt_box, convo])
     export_btn.click(fn=log_feedback_to_csv, inputs=[], outputs=[status_box])