|
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 |
