Update main.py

main.py

@@ -1,7 +1,12 @@
-import gspread
+from fastapi import FastAPI
+from pydantic import BaseModel
 from google.oauth2.service_account import Credentials
+import gspread
 from google.auth.exceptions import GoogleAuthError
 import os
+import uvicorn
+
+app = FastAPI()
 
 # Define chapter categories
 physics_chapters = {
@@ -29,7 +34,6 @@ maths_chapters = {
     "three dimensional geometry", "vector algebra", "statistics and probability", "trigonometry"
 }
 
-# Define base class for chapters
 class Chapter:
     def __init__(self):
         self.data = {}
@@ -41,12 +45,13 @@ class Chapter:
         self.data[chapter_name]["count"] += 1
 
     def print_averages(self):
+        result = {}
         for chapter, stats in self.data.items():
             if stats["count"] > 0:
                 avg = stats["sum_marks"] / stats["count"]
-                print(f"{chapter}: Average = {round(avg, 1)}")
+                result[chapter] = round(avg, 1)
+        return result
 
-# Create specific classes for Physics, Chemistry, and Mathematics
 class PhysicsChapter(Chapter):
     pass
 
@@ -56,7 +61,6 @@ class ChemistryChapter(Chapter):
 class MathsChapter(Chapter):
     pass
 
-# Function to fetch the credentials from environment variables or manually inputted credentials in Colab
 def get_credentials_from_env():
     service_account_info = {
         "type": os.getenv("SERVICE_ACCOUNT_TYPE"),
@@ -76,7 +80,7 @@ def get_credentials_from_env():
     creds = Credentials.from_service_account_info(service_account_info, scopes=scope)
     return creds
 
-# Main function to analyze chapter averages
+@app.get("/analyze_chapter_averages")
 def analyze_chapter_averages():
     try:
         # Set up credentials
@@ -118,20 +122,19 @@ def analyze_chapter_averages():
 
                 i += 2  # Move to the next chapter and marks
 
-        # Print average marks per chapter for each subject
-        print("Physics Chapters:")
-        physics_tracker.print_averages()
-        print("\nChemistry Chapters:")
-        chemistry_tracker.print_averages()
-        print("\nMaths Chapters:")
-        maths_tracker.print_averages()
+        # Return average marks per chapter for each subject
+        return {
+            "Physics Chapters": physics_tracker.print_averages(),
+            "Chemistry Chapters": chemistry_tracker.print_averages(),
+            "Maths Chapters": maths_tracker.print_averages()
+        }
 
     except GoogleAuthError as e:
-        print(f"Authentication error: {e}")
+        return {"error": f"Authentication error: {e}"}
     except gspread.exceptions.SpreadsheetNotFound:
-        print("Spreadsheet not found. Please check the URL.")
+        return {"error": "Spreadsheet not found. Please check the URL."}
     except Exception as e:
-        print(f"An error occurred: {e}")
+        return {"error": f"An error occurred: {e}"}
 
-# Call the main function
-analyze_chapter_averages()
+if __name__ == "__main__":
+    uvicorn.run(app, host="0.0.0.0", port=8000)