image save works, need to now put images on host.

Author: LoadingStill

Dockerfile

@@ -11,6 +11,10 @@ RUN pip install -r requirements.txt
 # Copy application code
 COPY . .
 
+# Make sure .env will work
+RUN pip install python-dotenv
+
+
 # Expose port and run Flask
 EXPOSE 5000
 CMD ["python3", "-m", "flask", "run", "--host=0.0.0.0"]

app.py

@@ -1,16 +1,29 @@
 from flask import Flask, render_template, request, jsonify, redirect, url_for
-from game_data import game_pages  # Import the game_pages dictionary
 import os
 import subprocess
+from game_data import game_pages
 
 app = Flask(__name__)
 
+# Absolute path to the image_downloader.py script in the Docker container
+image_downloader_path = os.path.join(os.path.dirname(__file__), 'static', 'images', 'image_downloader.py')
+
 @app.route('/')
 def home():
     return render_template('home.html', game_pages=game_pages)
 
-# Game list is generated at game_data.py
-# To add a new game make the html page and needed scripts first then add game at game_data.py last before publishing.
+@app.route('/start-download')
+def start_download():
+    try:
+        # Ensure that the script exists before running
+        if os.path.exists(image_downloader_path):
+            subprocess.run(['python', image_downloader_path], check=True)
+            return jsonify({"message": "Download started!"}), 200
+        else:
+            return jsonify({"message": "Image downloader script not found!"}), 404
+    except subprocess.CalledProcessError as e:
+        return jsonify({"message": f"Error: {e}"}), 500
+
 @app.route('/games/<game_name>.html')
 def game(game_name):
     if game_name in game_pages:

requirements.txt

@@ -2,38 +2,27 @@
 import json
 import requests
 
-# Path to the JSON file containing the game data
-json_file = 'image_urls.json'
+# Get the absolute path to the current directory (static/images)
+current_dir = os.path.dirname(__file__)
+json_file = os.path.join(current_dir, 'image_urls.json')  # This ensures the correct path
 
-# Folder where images will be saved
-save_folder = './static/images'
-
-# Ensure the save folder exists
-if not os.path.exists(save_folder):
-    os.makedirs(save_folder)
-
-# Function to download and save the image
-def download_image(image_url, save_path):
+# Check if the file exists
+if not os.path.exists(json_file):
+    print(f"File not found: {json_file}")
+else:
+    # Open and read the JSON file
     try:
-        # Send a GET request to the image URL
-        response = requests.get(image_url)
-        # Check if the request was successful
-        if response.status_code == 200:
-            # Open the file in write-binary mode and save the content
-            with open(save_path, 'wb') as file:
-                file.write(response.content)
-            print(f"Downloaded image and saved as {save_path}")
-        else:
-            print(f"Failed to download {image_url}, status code: {response.status_code}")
-    except Exception as e:
-        print(f"Error downloading {image_url}: {e}")
-
-# Load the JSON file containing the image URLs
-with open(json_file, 'r') as file:
-    game_data = json.load(file)
-
-# Iterate over each entry in the JSON data
-for game in game_data:
-    game_name, rename_to, image_url = game
-    save_path = os.path.join(save_folder, rename_to)
-    download_image(image_url, save_path)
+        with open(json_file, 'r') as file:
+            data = json.load(file)
+            print("Data loaded successfully")
+            # Add your code to download the images
+            for game, filename, url in data:
+                # Download the image
+                response = requests.get(url)
+                with open(os.path.join(current_dir, filename), 'wb') as img_file:
+                    img_file.write(response.content)
+                    print(f"Downloaded image and saved as {filename}")
+    except FileNotFoundError:
+        print(f"File not found: {json_file}")
+    except json.JSONDecodeError:
+        print("Error decoding JSON")

static/images/image_downloader.py

@@ -23,6 +23,21 @@ <h1>WebGSM</h1>
                 </a>
             {% endfor %}
         </div>
+        <!-- Add a button to start the image download process -->
+        <button id="download-button">Download Images</button>
+        <p id="download-status"></p>
     </div>
+
+    <script>
+        // When the download button is clicked
+        $('#download-button').click(function() {
+            // Send an AJAX request to start the download process
+            $.get('/start-download', function(data) {
+                $('#download-status').text(data.message);
+            }).fail(function(error) {
+                $('#download-status').text("Error starting the download.");
+            });
+        });
+    </script>
 </body>
 </html>