HemanM commited on
Commit
515b961
·
verified ·
1 Parent(s): 61203b9

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +18 -4
app.py CHANGED
@@ -1,7 +1,8 @@
1
  import gradio as gr
2
- from inference import evo_chat_predict, get_model_config, get_gpt_response, retrain_from_feedback
3
  import pandas as pd
4
  import csv
 
5
  from datetime import datetime
6
 
7
  feedback_log = []
@@ -55,10 +56,23 @@ with gr.Blocks(theme=gr.themes.Base(), css="body { background-color: #0f0f0f; co
55
 
56
  return evo_text, gpt_text, stats_text, updated_df
57
 
58
- # 🔁 Actual retraining
59
  def retrain_evo():
60
- status = retrain_from_feedback(feedback_log)
61
- return status
 
 
 
 
 
 
 
 
 
 
 
 
 
62
 
63
  # 🧹 Clear UI
64
  def clear_fields():
 
1
  import gradio as gr
2
+ from inference import evo_chat_predict, get_model_config, get_gpt_response
3
  import pandas as pd
4
  import csv
5
+ import os
6
  from datetime import datetime
7
 
8
  feedback_log = []
 
56
 
57
  return evo_text, gpt_text, stats_text, updated_df
58
 
59
+ # 🔁 Retrain Evo using retrain.py
60
  def retrain_evo():
61
+ if not feedback_log:
62
+ return "⚠️ No feedback data to retrain from."
63
+
64
+ # Save log to feedback_log.csv in required format
65
+ with open("feedback_log.csv", "w", newline="") as f:
66
+ writer = csv.writer(f)
67
+ writer.writerow(["prompt", "context", "label"])
68
+ for row in feedback_log:
69
+ prompt = row[0]
70
+ context = row[6] # context_used
71
+ label = 0 if row[3] == row[1] else 1 # 0 if answer == option1 else 1
72
+ writer.writerow([prompt, context, label])
73
+
74
+ os.system("python retrain.py")
75
+ return "✅ Evo retrained from feedback_log.csv."
76
 
77
  # 🧹 Clear UI
78
  def clear_fields():