Upload main.py

main.py

@@ -0,0 +1,69 @@
+from fastapi import FastAPI
+import gspread
+from google.oauth2.service_account import Credentials
+from google.auth.exceptions import GoogleAuthError
+from google.colab import userdata
+
+# 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": userdata.get("SERVICE_ACCOUNT_TYPE"),
+        "project_id": userdata.get("PROJECT_ID"),
+        "private_key_id": userdata.get("PRIVATE_KEY_ID"),
+        "private_key": userdata.get("PRIVATE_KEY").replace('\\n', '\n'),
+        "client_email": userdata.get("CLIENT_EMAIL"),
+        "client_id": userdata.get("CLIENT_ID"),
+        "auth_uri": userdata.get("AUTH_URI"),
+        "token_uri": userdata.get("TOKEN_URI"),
+        "auth_provider_x509_cert_url": userdata.get("AUTH_PROVIDER_X509_CERT_URL"),
+        "client_x509_cert_url": userdata.get("CLIENT_X509_CERT_URL"),
+        "universe_domain": userdata.get("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}"}