Update app.py

app.py

@@ -257,15 +257,14 @@ class AITrainerApp:
         total_texts = []
         for file_info in files:
             try:
-                file_path = file_info.name
-                with open(file_path, 'r', encoding='utf-8') as f:
-                    content = f.read()
-                    # Split into smaller chunks if needed
-                    chunks = self.split_into_chunks(content, 1000)
-                    total_texts.extend(chunks)
-                    self.output_log = self.log_output(f"Loaded {len(chunks)} chunks from {os.path.basename(file_path)}")
+                # Read the content from the file object
+                content = file_info.read().decode('utf-8')
+                # Split into smaller chunks if needed
+                chunks = self.split_into_chunks(content, 1000)
+                total_texts.extend(chunks)
+                self.output_log = self.log_output(f"Loaded {len(chunks)} chunks from {file_info.name}")
             except Exception as e:
-                error_msg = f"Error reading {file_path}: {str(e)}"
+                error_msg = f"Error reading {file_info.name}: {str(e)}"
                 self.output_log = self.log_output(error_msg)
                 return error_msg, self.output_log
         
@@ -300,8 +299,9 @@ class AITrainerApp:
         
     def start_training(self, d_model, n_layers, n_heads, batch_size, learning_rate, epochs, device_type):
         if not self.training_data:
-            self.output_log = self.log_output("Error: No training data loaded!")
-            return "Error: No training data loaded!", self.output_log, gr.update(interactive=False)
+            error_msg = "Error: No training data loaded!"
+            self.output_log = self.log_output(error_msg)
+            return error_msg, self.output_log, gr.update(interactive=True)
             
         self.stop_training_flag = False
         self.training_status = "Training started..."