| | import cv2 |
| | import numpy as np |
| | from nes_py import NESEnv |
| | from flask import Flask, Response, render_template_string |
| | import time |
| | import os |
| |
|
| | |
| | app = Flask(__name__) |
| |
|
| | |
| | running = False |
| | reset_flag = False |
| |
|
| | |
| | rom_path = "/home/ubuntu/mario/Super Mario Bros.nes" |
| |
|
| | |
| | try: |
| | env = NESEnv(rom_path) |
| | obs = env.reset() |
| | done = False |
| | print("NES environment initialized successfully with ROM:", rom_path) |
| | except Exception as e: |
| | print("Error initializing NES environment:", e) |
| | exit(1) |
| |
|
| | |
| | def generate_frames(): |
| | global running, reset_flag, done, obs |
| | frame_time = 1 / 30.0 |
| | print("Starting frame generation loop...") |
| | while running: |
| | start = time.time() |
| | if reset_flag: |
| | obs = env.reset() |
| | reset_flag = False |
| | done = False |
| | print("Game reset to title screen") |
| | if done: |
| | obs = env.reset() |
| | done = False |
| | print("Game reset due to done state") |
| | |
| | obs, reward, done, info = env.step(0) |
| | print("Step executed, Reward:", reward, "Done:", done) |
| | |
| | frame = env.get_image() |
| | if frame is None: |
| | print("Warning: Frame is None!") |
| | time.sleep(frame_time) |
| | continue |
| | frame = cv2.cvtColor(frame, cv2.COLOR_RGB2BGR) |
| | print("Captured frame shape:", frame.shape) |
| | |
| | ret, buffer = cv2.imencode('.jpg', frame) |
| | if not ret: |
| | print("Warning: Failed to encode frame!") |
| | time.sleep(frame_time) |
| | continue |
| | frame_bytes = buffer.tobytes() |
| | yield (b'--frame\r\n' |
| | b'Content-Type: image/jpeg\r\n\r\n' + frame_bytes + b'\r\n') |
| | elapsed = time.time() - start |
| | if elapsed < frame_time: |
| | time.sleep(frame_time - elapsed) |
| | print("Frame sent, elapsed time:", elapsed) |
| |
|
| | |
| | @app.route('/video_feed') |
| | def video_feed(): |
| | if not running: |
| | return "Stream not running", 503 |
| | return Response(generate_frames(), mimetype='multipart/x-mixed-replace; boundary=frame') |
| |
|
| | |
| | @app.route('/') |
| | def index(): |
| | return render_template_string(''' |
| | <!DOCTYPE html> |
| | <html> |
| | <head> |
| | <title>Mario Title Screen Stream</title> |
| | <style> |
| | body { background: #1e1e1e; color: white; font-family: Arial, sans-serif; text-align: center; } |
| | img { border: 2px solid #00ff00; } |
| | </style> |
| | </head> |
| | <body> |
| | <h1>Mario Title Screen Live Stream</h1> |
| | <img src="/video_feed" width="256" height="240"> |
| | </body> |
| | </html> |
| | ''') |
| |
|
| | |
| | if __name__ == "__main__": |
| | print("Starting Flask server on http://0.0.0.0:5111...") |
| | running = True |
| | try: |
| | app.run(host='0.0.0.0', port=5111, debug=True, use_reloader=False) |
| | except KeyboardInterrupt: |
| | print("Stopping the stream...") |
| | running = False |
