Update google_drive_read_preprompt.py

tabs/google_drive_read_preprompt.py

@@ -39,14 +39,20 @@ access_token = get_access_token()
 
 # Fonction pour lire le contenu d'un Google Doc
 def read_google_doc(doc_id):
-    # Créer les credentials à partir du token d'accès
-    creds = Credentials(token=access_token)
-
-    # Construire le service Google Docs avec les credentials
-    docs_service = build('docs', 'v1', credentials=creds)
-
-    # Requête pour obtenir le contenu du document
     try:
+        # Créer les credentials à partir du token d'accès
+        creds = Credentials(
+            token=access_token,
+            refresh_token=REFRESH_TOKEN,
+            token_uri="https://oauth2.googleapis.com/token",
+            client_id=CLIENT_ID,
+            client_secret=CLIENT_SECRET
+        )
+        
+        # Construire le service Google Docs avec les credentials
+        docs_service = build('docs', 'v1', credentials=creds)
+
+        # Requête pour obtenir le contenu du document
         document = docs_service.documents().get(documentId=doc_id).execute()
         content = document.get('body', {}).get('content', [])
 
@@ -56,16 +62,15 @@ def read_google_doc(doc_id):
         # Extraire chaque ligne de texte
         for element in content:
             if 'paragraph' in element:
-                paragraph_text = ""  # Regrouper tout le texte d'un paragraphe
+                paragraph_text = ""
                 for paragraph in element['paragraph']['elements']:
                     t = paragraph.get('textRun', {}).get('content', '')
-                    paragraph_text += t  # Ajouter le texte à la ligne du paragraphe
+                    paragraph_text += t
 
-                # Ajouter le paragraphe complet s'il contient du texte
                 if paragraph_text.strip():
                     lines.append(paragraph_text.strip())
 
-        return lines  # Retourne une liste contenant toutes les lignes
+        return lines
     except Exception as e:
         raise Exception(f"Erreur lors de la lecture du document : {e}")
 
@@ -84,30 +89,33 @@ def read_param():
 def format_param():
     try:
         lines = read_param()
+        if not lines:
+            raise Exception("Le contenu du document est vide.")
+        
         label = []
         question = []
         options = [[] for _ in range(8)]
         i = 0
         for p in range(8):
-            while (lines[i][:3] == "==="):
-                i +=1
+            while i < len(lines) and len(lines[i]) >= 3 and lines[i][:3] == "===":
+                i += 1
             label.append(lines[i][8:])
-            i +=1
-            if p not in [0,1,2]:
+            i += 1
+            if p not in [0, 1, 2]:
                 question.append(lines[i])
-                i +=1
-            else: question.append("")
-            while (lines[i][:3] != "==="):
+                i += 1
+            else:
+                question.append("")
+            while i < len(lines) and len(lines[i]) >= 3 and lines[i][:3] != "===":
                 options[p].append(lines[i])
-                i +=1
-        i+=1
-        while i<len(lines):
+                i += 1
+        i += 1
+        while i < len(lines):
             question.append(lines[i])
-            i +=1
-        print("label:\n",label)
-        print("question:\n",question)
-        print("options:\n",options)
+            i += 1
+        print("label:\n", label)
+        print("question:\n", question)
+        print("options:\n", options)
     except Exception as e:
         st.write(f"Erreur : {e}")
-    return label, question, options           
-        
+    return label, question, options