Update main.py

main.py

@@ -34,6 +34,21 @@ maths_chapters = {
     "three dimensional geometry", "vector algebra", "statistics and probability", "trigonometry"
 }
 
+# class Chapter:
+#     def __init__(self):
+#         self.data = {}
+
+#     def add_marks(self, chapter_name, marks):
+#         if chapter_name not in self.data:
+#             self.data[chapter_name] = {"sum_marks": 0, "count": 0}
+#         self.data[chapter_name]["sum_marks"] += marks
+#         self.data[chapter_name]["count"] += 1
+
+#     def print_averages(self):
+#         for chapter, stats in self.data.items():
+#             if stats["count"] > 0:
+#                 avg = stats["sum_marks"] / stats["count"]
+#         return f"{chapter}: Average = {round(avg, 1)}"
 class Chapter:
     def __init__(self):
         self.data = {}
@@ -45,10 +60,15 @@ class Chapter:
         self.data[chapter_name]["count"] += 1
 
     def print_averages(self):
+        averages = {}
         for chapter, stats in self.data.items():
             if stats["count"] > 0:
                 avg = stats["sum_marks"] / stats["count"]
-        return f"{chapter}: Average = {round(avg, 1)}"
+                averages[chapter] = round(avg, 1)
+            else:
+                averages[chapter] = 0
+        return averages
+
 
 class PhysicsChapter(Chapter):
     pass