Update src/services/journal_engine.py

src/services/journal_engine.py

@@ -3,7 +3,8 @@ import io
 import os
 import re
 from googleapiclient.discovery import build
-from googleapiclient.http import MediaByteArrayUpload, MediaIoBaseDownload
+# [FIX] Import MediaIoBaseUpload instead of the non-existent MediaByteArrayUpload
+from googleapiclient.http import MediaIoBaseUpload, MediaIoBaseDownload
 from google.oauth2.credentials import Credentials
 from ..config import get_user_keys, update_user_keys
 
@@ -35,17 +36,45 @@ class JournalEngine:
     @staticmethod
     def save_to_drive(service, file_id, journal_data):
         """Uploads the updated journal list back to the hidden AppData folder."""
-        media = MediaByteArrayUpload(
-            json.dumps(journal_data).encode('utf-8'), 
-            mimetype='application/json'
+        # [FIX] Use MediaIoBaseUpload with a BytesIO stream
+        media = MediaIoBaseUpload(
+            io.BytesIO(json.dumps(journal_data).encode('utf-8')), 
+            mimetype='application/json',
+            resumable=True
         )
         service.files().update(fileId=file_id, media_body=media).execute()
 
+    @staticmethod
+    def initialize_journal(service):
+        """Finds or creates 'journal.json' in the appDataFolder."""
+        response = service.files().list(
+            spaces='appDataFolder',
+            fields='files(id, name)',
+            pageSize=1
+        ).execute()
+        
+        files = response.get('files', [])
+        if files:
+            return files[0]['id']
+        
+        # Create it if it doesn't exist
+        file_metadata = {
+            'name': 'journal.json',
+            'parents': ['appDataFolder']
+        }
+        # [FIX] Use MediaIoBaseUpload here too
+        media = MediaIoBaseUpload(
+            io.BytesIO(json.dumps([]).encode('utf-8')), 
+            mimetype='application/json',
+            resumable=True
+        )
+        file = service.files().create(body=file_metadata, media_body=media, fields='id').execute()
+        return file.get('id')
+
     @staticmethod
     def parse_pnl(pnl_str):
         """Helper to extract a numeric value from strings like '+$500' or '-10%'."""
         try:
-            # Remove currency symbols and find the first number
             clean = re.sub(r'[^\d\.-]', '', str(pnl_str))
             return float(clean) if clean else 0.0
         except ValueError:
@@ -63,7 +92,6 @@ class JournalEngine:
         winrate = (len(wins) / total) * 100 if total > 0 else 0
         best_trade = max(journal_data, key=lambda x: cls.parse_pnl(x.get('pnl', 0)), default={})
         
-        # Identify the most frequent bias tag
         biases = [t.get('bias', 'Neutral') for t in journal_data if t.get('bias')]
         main_bias = max(set(biases), key=biases.count) if biases else "Neutral"