Update app.py

app.py

@@ -1,31 +1,35 @@
 # app.py
 from flask import Flask, send_from_directory, jsonify, request
 import os
+import json
 
 app = Flask(__name__)
 
-# Updated simulation parameters with real-world masses (scaled down for simulation)
-# Sun: 1.989 x 10^30 kg, Earth: 5.972 x 10^24 kg, Mars: 6.417 x 10^23 kg
-# Scale factor: 1e-26 for manageable numbers
+# Directory to store settings
+SETTINGS_DIR = "settings"
+if not os.path.exists(SETTINGS_DIR):
+    os.makedirs(SETTINGS_DIR)
+
+# Real-world simulation parameters (unscaled)
 simulation_params = {
     "sun": {
-        "mass": 1.989e4,  # Scaled mass (1.989 x 10^30 kg * 1e-26)
+        "mass": 1.989e30,  # Real mass in kg
         "position": [0, 0, 0],
         "orbital_velocity": 0,  # Sun is stationary
     },
     "earth": {
-        "mass": 5.972e-2,  # Scaled mass (5.972 x 10^24 kg * 1e-26)
-        "position": [10, 0, 0],  # Approx 1 AU scaled down (149.6 million km -> 10 units)
-        "orbital_velocity": 0.0298,  # 29.8 km/s scaled down
+        "mass": 5.972e24,  # Real mass in kg
+        "position": [149.6e6, 0, 0],  # 1 AU in km
+        "orbital_velocity": 29.8,  # Real orbital velocity in km/s
     },
     "mars": {
-        "mass": 6.417e-3,  # Scaled mass (6.417 x 10^23 kg * 1e-26)
-        "position": [15, 0, 0],  # Approx 1.52 AU scaled down (227.9 million km -> 15 units)
-        "orbital_velocity": 0.0241,  # 24.1 km/s scaled down
+        "mass": 6.417e23,  # Real mass in kg
+        "position": [227.9e6, 0, 0],  # 1.52 AU in km
+        "orbital_velocity": 24.1,  # Real orbital velocity in km/s
     },
     "fluid_speed": 0.1,
-    "fluid_friction": 0.9,  # Default friction factor
-    "fluid_deflection": 0.1,  # Default deflection strength
+    "fluid_friction": 0.9,
+    "fluid_deflection": 0.1,
 }
 
 # Serve the frontend
@@ -51,6 +55,32 @@ def update_params():
     simulation_params.update(data)
     return jsonify({"status": "success", "params": simulation_params})
 
+# API to save settings to a JSON file
+@app.route('/api/save', methods=['POST'])
+def save_settings():
+    try:
+        filename = os.path.join(SETTINGS_DIR, "settings.json")
+        with open(filename, 'w') as f:
+            json.dump(simulation_params, f, indent=4)
+        return jsonify({"status": "success", "message": "Settings saved successfully"})
+    except Exception as e:
+        return jsonify({"status": "error", "message": str(e)}), 500
+
+# API to load settings from a JSON file
+@app.route('/api/load', methods=['GET'])
+def load_settings():
+    global simulation_params
+    try:
+        filename = os.path.join(SETTINGS_DIR, "settings.json")
+        if os.path.exists(filename):
+            with open(filename, 'r') as f:
+                simulation_params = json.load(f)
+            return jsonify({"status": "success", "params": simulation_params})
+        else:
+            return jsonify({"status": "error", "message": "No saved settings found"}), 404
+    except Exception as e:
+        return jsonify({"status": "error", "message": str(e)}), 500
+
 if __name__ == '__main__':
     port = int(os.environ.get('PORT', 7860))  # Default port for Hugging Face Spaces
     app.run(host='0.0.0.0', port=port)