Update app.py

app.py

@@ -8,11 +8,15 @@ BASE_DIR = "./brevet"
 # Utility function to list files for a given subject and type
 def get_files(subject, file_type):
     subject_path = os.path.join(BASE_DIR, subject)
-    files = [f.replace(' Corrigé', '') for f in os.listdir(subject_path) if 'Corrigé' in f] if file_type == 'Corrigé' else [f for f in os.listdir(subject_path) if 'Corrigé' not in f]
+    # Adjust to remove '.pdf' from filenames
+    if file_type == 'Corrigé':
+        files = [f.replace(' Corrigé.pdf', '') for f in os.listdir(subject_path) if 'Corrigé' in f]
+    else:
+        files = [f.replace('.pdf', '') for f in os.listdir(subject_path) if 'Corrigé' not in f and f.endswith('.pdf')]
     return sorted(files, reverse=True)
 
 # Streamlit app layout
-st.title("Sujets et Corrigés du Brevet (*Par Studyrama*)")
+st.title("Téléchargement des sujets et corrigés du Brevet")
 
 # Subject selection
 subject = st.selectbox("Sélectionner la matière", ["Francais", "Mathématique", "Science", "Histoire-Géographie-EMC"])
@@ -24,8 +28,7 @@ file_type = st.selectbox("Sélectionner le type", ["Sujet", "Corrigé"])
 files = get_files(subject, file_type)
 selected_file = st.selectbox("Sélectionner l'année", files)
 
-selected_file_corrected = selected_file.replace('.pdf', '')
-file_name = f"{selected_file_corrected}{' Corrigé' if file_type == 'Corrigé' else ''}.pdf"
+file_name = f"{selected_file}{' Corrigé' if file_type == 'Corrigé' else ''}.pdf"
 file_path = os.path.join(BASE_DIR, subject, file_name)
 
 # Initialize or update session state for button press