Update wltv_server.py

wltv_server.py

@@ -28,6 +28,11 @@ def extract_file(zip_path, extract_to_dir):
 # Step 3: Set up a Flask route to expose the extracted file
 @app.route('/wltv', methods=['GET'])
 def serve_extracted_file():
+    # Ensure the extracted directory exists
+    if not os.path.exists('extracted'):
+        logger.error("Extracted directory does not exist. Creating it now...")
+        os.makedirs('extracted')
+    
     # Log contents of extracted directory for debugging
     extracted_files = os.listdir('extracted')
     logger.info(f"Files in 'extracted' directory: {extracted_files}")
@@ -54,4 +59,4 @@ if __name__ == '__main__':
     extract_file(zip_path, extract_to_dir)
 
     # Start the Flask web server
-    app.run(host='0.0.0.0', port=8080)
+    app.run(host='0.0.0.0', port=8080)