Update app.py

app.py

@@ -342,6 +342,7 @@ def remove_files():
     return redirect(url_for('index'))
 
 # API for downloading the latest trained model
+'''
 @app.route('/download_model', methods=['GET'])
 def download_latest_model():
     try:
@@ -374,6 +375,49 @@ def download_latest_model():
     except Exception as e:
         flash(f"Error while downloading the model: {str(e)}", 'error')
         return redirect(request.referrer)
+'''
+@app.route('/download_model', methods=['GET'])
+def download_latest_model():
+    try:
+        models_dir = app.config['MODELS_FOLDER']
+        model_files = os.listdir(models_dir)
+
+        if not model_files:
+            flash('No model files found', 'error')
+            return redirect(request.referrer)
+
+        # Sort model files and get the latest one
+        latest_model_file = sorted(model_files, reverse=True)[0]
+        
+        # Full path to the latest model file
+        model_path = os.path.join(models_dir, latest_model_file)
+        
+        if not os.path.exists(model_path):
+            flash('Model file not found on the server', 'error')
+            return redirect(request.referrer)
+        
+        # Create a zip file with the model
+        zip_filename = os.path.join(models_dir, f"{latest_model_file}.zip")
+        
+        # Create a zip file that includes the whole directory structure of the model
+        with zipfile.ZipFile(zip_filename, 'w', zipfile.ZIP_DEFLATED) as zipf:
+            # If the model is a directory, zip the whole directory
+            if os.path.isdir(model_path):
+                for foldername, subfolders, filenames in os.walk(model_path):
+                    for filename in filenames:
+                        file_path = os.path.join(foldername, filename)
+                        # Write the file, preserving the directory structure
+                        zipf.write(file_path, os.path.relpath(file_path, models_dir))
+            else:
+                # If it's just a single file, zip it
+                zipf.write(model_path, os.path.basename(model_path))
+        
+        # Send the zip file as a download
+        return send_file(zip_filename, as_attachment=True)
+
+    except Exception as e:
+        flash(f"Error while downloading the model: {str(e)}", 'error')
+        return redirect(request.referrer)
         
 # Route to serve uploaded files
 @app.route('/<foldername>/zip', methods=['GET'])