Chirag0123 commited on
Commit
1dd1900
Β·
1 Parent(s): 0b0338d

fix(gradio): Share env state with FastAPI to fix 3D Visualizer data tracking

Browse files
Files changed (1) hide show
  1. app.py +16 -22
app.py CHANGED
@@ -20,32 +20,26 @@ app.py β€” Gradio UI v4.0 β€” Full Research Platform
20
  import os
21
  import json
22
  import gradio as gr
23
- from server.environment import CodebaseNavEnvironment
 
 
 
 
 
 
 
 
 
 
 
 
 
24
  from server.models import RepoAction
25
- from server.failure_classifier import FailureClassifier
26
- from server.strategy_detector import StrategyDetector
27
- from server.advanced_metrics import AdvancedMetricsEngine
28
- from server.self_improvement import SelfImprovementEngine
29
- from server.multi_agent import MultiAgentComparison
30
- from server.causal_probe import CausalProbe
31
- from server.counterfactual_engine import CounterfactualEngine
32
- from server.confidence_calibrator import ConfidenceCalibrator
33
- from server.benchmark_runner import BenchmarkRunner
34
- from server.analytics_engine import AnalyticsEngine
35
  from server.memory_bank import get_global_memory
36
 
37
  # ── Global instances ──────────────────────────────────────────────────────────
38
- env = CodebaseNavEnvironment()
39
- failure_clf = FailureClassifier()
40
- strategy_det = StrategyDetector()
41
- adv_metrics_engine = AdvancedMetricsEngine()
42
- improvement_engine = SelfImprovementEngine()
43
- multi_agent_engine = MultiAgentComparison()
44
- causal_probe = CausalProbe()
45
- counterfactual_engine = CounterfactualEngine()
46
- confidence_calibrator = ConfidenceCalibrator()
47
- benchmark_runner = BenchmarkRunner()
48
- analytics_engine = AnalyticsEngine()
49
  memory_bank = get_global_memory()
50
 
51
 
 
20
  import os
21
  import json
22
  import gradio as gr
23
+ from server.app import (
24
+ app as fastapi_app,
25
+ env,
26
+ failure_clf,
27
+ strategy_det,
28
+ adv_metrics as adv_metrics_engine,
29
+ improvement as improvement_engine,
30
+ multi_agent as multi_agent_engine,
31
+ _causal as causal_probe,
32
+ _counter as counterfactual_engine,
33
+ _calibrator as confidence_calibrator,
34
+ _benchmark as benchmark_runner,
35
+ _analytics as analytics_engine,
36
+ )
37
  from server.models import RepoAction
 
 
 
 
 
 
 
 
 
 
38
  from server.memory_bank import get_global_memory
39
 
40
  # ── Global instances ──────────────────────────────────────────────────────────
41
+ # All engines and the environment are imported from server.app so that
42
+ # Gradio interactions and direct HTTP REST calls use the exact same state.
 
 
 
 
 
 
 
 
 
43
  memory_bank = get_global_memory()
44
 
45