Spaces:
Sleeping
Sleeping
| import gradio as gr | |
| import os | |
| import random | |
| from evaluate import run_evaluation | |
| from visualize import generate_graph | |
| def run_simulation(manual_seed=None): | |
| try: | |
| seed = int(manual_seed) if manual_seed and str(manual_seed).isdigit() else random.randint(1000, 99999) | |
| import io | |
| from contextlib import redirect_stdout | |
| f = io.StringIO() | |
| with redirect_stdout(f): | |
| scores = run_evaluation(base_seed=seed, silent=False) | |
| logs = f.getvalue() | |
| graph_path = "optimization_results.png" | |
| generate_graph(scores, seed, output_path=graph_path) | |
| return logs, graph_path | |
| except Exception as e: | |
| return f"Error running simulation: {str(e)}", None | |
| with gr.Blocks() as interface: | |
| gr.Markdown("# 🚦 Smart Traffic Optimization Environment (OpenEnv)") | |
| gr.Markdown("Welcome to the Traffic Simulator. Watch how our AI improves traffic flow in real time.") | |
| with gr.Row(): | |
| with gr.Column(scale=1): | |
| pass | |
| with gr.Column(scale=2, min_width=320): | |
| seed_input = gr.Textbox(label="Optional Seed (Empty for Random)", placeholder="e.g. 42") | |
| run_btn = gr.Button("🚀 Run Traffic Evaluator", variant="primary", size="lg") | |
| gr.HTML(""" | |
| <p style=' | |
| text-align: center; | |
| color: rgba(100, 100, 100, 0.75); | |
| font-size: 0.88em; | |
| margin: 4px 0 16px 0; | |
| font-weight: 400; | |
| letter-spacing: 0.01em; | |
| '>Click to simulate AI-driven traffic optimization across difficulty levels.</p> | |
| """) | |
| with gr.Column(scale=1): | |
| pass | |
| gr.Markdown(""" | |
| <div style='background-color: #ffeaea; border-left: 4px solid #ff4d4f; padding: 12px; margin: 15px 0px; border-radius: 4px; box-shadow: 0 1px 3px rgba(0,0,0,0.1);'> | |
| <span style='color: #a8071a; font-weight: 600; font-size: 1.05em;'>🚨 Active Protocol:</span> | |
| <span style='color: #434343;'>System prioritizes emergency vehicles in real-time.</span> | |
| </div> | |
| """) | |
| with gr.Row(): | |
| with gr.Column(scale=1): | |
| gr.Markdown("<h3 style='text-align: center; color: #555; margin-bottom: 2px;'>Simulation Logs</h3>") | |
| output_text = gr.Textbox(show_label=False, lines=22, interactive=False) | |
| with gr.Column(scale=1): | |
| gr.Markdown("<h3 style='text-align: center; color: #222; margin-bottom: 10px; border-bottom: 1px solid #eaeaea; padding-bottom: 5px;'>Performance Improvement After Optimization</h3>") | |
| output_img = gr.Image(show_label=False, type="filepath") | |
| run_btn.click(fn=run_simulation, inputs=[seed_input], outputs=[output_text, output_img]) | |
| if __name__ == "__main__": | |
| interface.launch(server_name="0.0.0.0", server_port=7860, theme=gr.themes.Soft()) | |