Update wltv_server.py

wltv_server.py

@@ -1,31 +1,43 @@
 import os
 import requests
 import zipfile
+import logging
 from flask import Flask, send_file
 
+# Set up logging
+logging.basicConfig(level=logging.INFO)
+logger = logging.getLogger(__name__)
+
 app = Flask(__name__)
 
 # Step 1: Download the zip file
 def download_zip_file(url, output_path):
+    logger.info(f"Downloading from {url}...")
     response = requests.get(url)
     with open(output_path, 'wb') as f:
         f.write(response.content)
-    print(f"Downloaded zip file to {output_path}")
+    logger.info(f"Downloaded zip file to {output_path}")
 
 # Step 2: Extract the file from the zip file
 def extract_file(zip_path, extract_to_dir):
+    logger.info(f"Extracting {zip_path} to {extract_to_dir}...")
     with zipfile.ZipFile(zip_path, 'r') as zip_ref:
         zip_ref.extractall(extract_to_dir)
-    print(f"Extracted zip file to {extract_to_dir}")
+    logger.info(f"Extracted zip file to {extract_to_dir}")
 
 # Step 3: Set up a Flask route to expose the extracted file
 @app.route('/wltv', methods=['GET'])
 def serve_extracted_file():
-    # Assuming the file is named "it_wltv_full" inside the extracted directory
+    # Log contents of extracted directory for debugging
+    extracted_files = os.listdir('extracted')
+    logger.info(f"Files in 'extracted' directory: {extracted_files}")
+    
     file_path = os.path.join('extracted', 'it_wltv_full')
     if os.path.exists(file_path):
+        logger.info(f"Serving file from {file_path}")
         return send_file(file_path, as_attachment=False, mimetype='text/plain')
     else:
+        logger.error("File not found")
         return "File not found", 404
 
 if __name__ == '__main__':