Anurag Prasad commited on
Commit
f7527e8
·
1 Parent(s): d65ad43

removed duplicates

Browse files
Files changed (1) hide show
  1. app.py +2 -16
app.py CHANGED
@@ -197,17 +197,6 @@ def save_model_to_db(model_name: str, system_prompt: str):
197
  print(f"Error saving model: {e}")
198
  return f"Error saving model: {e}"
199
 
200
- def calculate_drift_via_mcp(model_name: str):
201
- """Calculate drift for model via MCP"""
202
- try:
203
- result = run_async(mcp_client.call_tool("calculate_drift", {"model_name": model_name}))
204
- return result
205
- except Exception as e:
206
- print(f"Error calculating drift: {e}")
207
- import random
208
- drift_score = round(random.uniform(0.05, 0.25), 3)
209
- return {"drift_score": drift_score, "message": f"Drift calculated and saved for {model_name}"}
210
-
211
  def get_drift_history_from_db(model_name: str):
212
  """Get drift history from database via MCP"""
213
  try:
@@ -277,10 +266,6 @@ def on_model_select(dropdown_value):
277
  actual_model_name = extract_model_name_from_dropdown(dropdown_value, current_model_mapping)
278
  return actual_model_name, actual_model_name
279
 
280
- def toggle_create_new():
281
- """Toggle create new model section visibility"""
282
- return gr.update(visible=True)
283
-
284
  def cancel_create_new():
285
  """Cancel create new model"""
286
  return [
@@ -375,9 +360,10 @@ def calculate_drift(dropdown_value):
375
  return "\n".join(msg.text for msg in result)
376
  except Exception as e:
377
  print(f"Error calculating drift: {e}")
 
378
 
379
  # Fallback to the simpler drift calculation if needed
380
- result = calculate_drift_handler({"model_name": model_name})
381
  return f"Drift Score: {result.get('drift_score', 0.0):.3f}\n{result.get('message', '')}"
382
 
383
  def refresh_drift_history(dropdown_value):
 
197
  print(f"Error saving model: {e}")
198
  return f"Error saving model: {e}"
199
 
 
 
 
 
 
 
 
 
 
 
 
200
  def get_drift_history_from_db(model_name: str):
201
  """Get drift history from database via MCP"""
202
  try:
 
266
  actual_model_name = extract_model_name_from_dropdown(dropdown_value, current_model_mapping)
267
  return actual_model_name, actual_model_name
268
 
 
 
 
 
269
  def cancel_create_new():
270
  """Cancel create new model"""
271
  return [
 
360
  return "\n".join(msg.text for msg in result)
361
  except Exception as e:
362
  print(f"Error calculating drift: {e}")
363
+ return f"Error calculating drift from server side: {e}"
364
 
365
  # Fallback to the simpler drift calculation if needed
366
+ # result = calculate_drift_handler({"model_name": model_name})
367
  return f"Drift Score: {result.get('drift_score', 0.0):.3f}\n{result.get('message', '')}"
368
 
369
  def refresh_drift_history(dropdown_value):