Update main.py

main.py

@@ -1,3 +1,72 @@
+# from fastapi import FastAPI
+# import gspread
+# from google.oauth2.service_account import Credentials
+# from google.auth.exceptions import GoogleAuthError
+# import os
+
+# # Define the Academic and Non-Academic panic buttons
+# academic_panic_buttons = ["MISSED CLASSES", "BACKLOGS", "LACK OF MOTIVATION", "NOT UNDERSTANDING", "BAD MARKS"]
+# non_academic_panic_buttons = ["EMOTIONAL FACTORS", "PROCRASTINATE", "LOST INTEREST", "LACK OF FOCUS", "GOALS NOT ACHIEVED", "LACK OF DISCIPLINE"]
+
+# app = FastAPI()
+
+# # Function to fetch the credentials
+# def get_credentials_from_env():
+#     service_account_info = {
+#         "type": os.getenv("SERVICE_ACCOUNT_TYPE"),
+#         "project_id": os.getenv("PROJECT_ID"),
+#         "private_key_id": os.getenv("PRIVATE_KEY_ID"),
+#         "private_key": os.getenv("PRIVATE_KEY").replace('\\n', '\n'),
+#         "client_email": os.getenv("CLIENT_EMAIL"),
+#         "client_id": os.getenv("CLIENT_ID"),
+#         "auth_uri": os.getenv("AUTH_URI"),
+#         "token_uri": os.getenv("TOKEN_URI"),
+#         "auth_provider_x509_cert_url": os.getenv("AUTH_PROVIDER_X509_CERT_URL"),
+#         "client_x509_cert_url": os.getenv("CLIENT_X509_CERT_URL"),
+#         "universe_domain": os.getenv("UNIVERSE_DOMAIN")
+#     }
+    
+#     scope = ['https://spreadsheets.google.com/feeds', 'https://www.googleapis.com/auth/drive']
+#     creds = Credentials.from_service_account_info(service_account_info, scopes=scope)
+#     return creds
+
+# # Main function to fetch the panic button occurrences
+# @app.get("/panic-button-occurances")
+# def get_panic_button_occurrences():
+#     try:
+#         # Set up credentials
+#         creds = get_credentials_from_env()
+#         client = gspread.authorize(creds)
+
+#         # Open the Google Sheet
+#         sheet = client.open_by_url('https://docs.google.com/spreadsheets/d/1nFZGkCvRV6qS-mhsORhX3dxI0JSge32_UwWgWKl3eyw/edit?gid=0#gid=0').worksheet('Sheet1')
+
+#         # Get all values from the sheet
+#         data = sheet.get_all_values()
+
+#         # Initialize the lists with panic button names and 0 counts
+#         academic_list = {button: 0 for button in academic_panic_buttons}
+#         non_academic_list = {button: 0 for button in non_academic_panic_buttons}
+
+#         # Iterate through all rows in the data
+#         for row in data:
+#             panic_button = row[1]  # Assuming the panic button values are in the second column
+#             if panic_button in academic_list:
+#                 academic_list[panic_button] += 1
+#             elif panic_button in non_academic_list:
+#                 non_academic_list[panic_button] += 1
+
+#         return {
+#             "academic": academic_list,
+#             "non_academic": non_academic_list
+#         }
+
+#     except GoogleAuthError as e:
+#         return {"error": f"Authentication error: {e}"}
+#     except gspread.exceptions.SpreadsheetNotFound:
+#         return {"error": "Spreadsheet not found. Please check the URL."}
+#     except Exception as e:
+#         return {"error": f"An error occurred: {e}"}
 from fastapi import FastAPI
 import gspread
 from google.oauth2.service_account import Credentials
@@ -50,11 +119,16 @@ def get_panic_button_occurrences():
 
         # Iterate through all rows in the data
         for row in data:
-            panic_button = row[1:]  # Assuming the panic button values are in the second column
-            if panic_button in academic_list:
-                academic_list[panic_button] += 1
-            elif panic_button in non_academic_list:
-                non_academic_list[panic_button] += 1
+            for panic_button in row:
+                # Stop when an empty column is encountered
+                if not panic_button:
+                    break
+
+                # Check and update counts for academic and non-academic panic buttons
+                if panic_button in academic_list:
+                    academic_list[panic_button] += 1
+                elif panic_button in non_academic_list:
+                    non_academic_list[panic_button] += 1
 
         return {
             "academic": academic_list,