Spaces:
Running
Running
| import os | |
| import requests | |
| import streamlit as st | |
| from googleapiclient.discovery import build | |
| from google.auth.credentials import AnonymousCredentials | |
| from google.auth.transport.requests import Request | |
| from dotenv import load_dotenv | |
| from google.oauth2.credentials import Credentials | |
| if st.session_state.Cloud != 0: | |
| load_dotenv() | |
| # Charger les secrets depuis les variables d'environnement | |
| CLIENT_ID = os.getenv("GOOGLE_CLIENT_ID") | |
| CLIENT_SECRET = os.getenv("GOOGLE_CLIENT_SECRET") | |
| REFRESH_TOKEN = os.getenv("GOOGLE_REFRESH_TOKEN") | |
| GOOGLE_PREPROMPT_FILE_ID = os.getenv("GOOGLE_PREPROMPT_FILE_ID") | |
| # URL pour rafraîchir le token | |
| TOKEN_URL = "https://oauth2.googleapis.com/token" | |
| # Fonction pour obtenir un token d'accès | |
| def get_access_token(): | |
| data = { | |
| "client_id": CLIENT_ID, | |
| "client_secret": CLIENT_SECRET, | |
| "refresh_token": REFRESH_TOKEN, | |
| "grant_type": "refresh_token", | |
| } | |
| response = requests.post(TOKEN_URL, data=data) | |
| response_data = response.json() | |
| if "access_token" in response_data: | |
| return response_data["access_token"] | |
| else: | |
| raise Exception(f"Erreur d'obtention du token d'accès : {response_data}") | |
| # Obtenir le token d'accès | |
| access_token = get_access_token() | |
| # Fonction pour lire le contenu d'un Google Doc | |
| def read_google_doc(doc_id): | |
| 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', []) | |
| # Initialiser une liste pour stocker chaque ligne | |
| lines = [] | |
| # Extraire chaque ligne de texte | |
| for element in content: | |
| if 'paragraph' in element: | |
| paragraph_text = "" | |
| for paragraph in element['paragraph']['elements']: | |
| t = paragraph.get('textRun', {}).get('content', '') | |
| paragraph_text += t | |
| if paragraph_text.strip(): | |
| lines.append(paragraph_text.strip()) | |
| return lines | |
| except Exception as e: | |
| raise Exception(f"Erreur lors de la lecture du document : {e}") | |
| def read_param(): | |
| # Lire et afficher le contenu d'un fichier | |
| try: | |
| lines = read_google_doc(GOOGLE_PREPROMPT_FILE_ID) | |
| return lines | |
| except Exception as e: | |
| st.write(f"Erreur : {e}") | |
| 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 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]: | |
| question.append(lines[i]) | |
| 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): | |
| question.append(lines[i]) | |
| i += 1 | |
| except Exception as e: | |
| st.write(f"Erreur : {e}") | |
| return label, question, options |