Atualização do app.py com função extract_tm_info

app.py

@@ -5,6 +5,18 @@ import pandas as pd
 from tensorflow.keras.models import model_from_json
 from collections import Counter
 
+def extract_tm_info(tm_path):
+    with zipfile.ZipFile(tm_path, 'r') as zip_ref:
+        with zip_ref.open('manifest.json') as f:
+            manifest = json.load(f)
+        return {
+            'type': manifest.get('type', 'N/A'),
+            'version': manifest.get('version', 'N/A'),
+            'epochs': manifest.get('appdata', {}).get('trainEpochs', 'N/A'),
+            'batch_size': manifest.get('appdata', {}).get('trainBatchSize', 'N/A'),
+            'learning_rate': manifest.get('appdata', {}).get('trainLearningRate', 'N/A')
+        }
+
 def extract_zip_info(zip_path):
     with zipfile.ZipFile(zip_path, 'r') as zip_ref:
         file_list = zip_ref.namelist()