Fixed os dir path

AlgoArchives · Jun 8, 2024 · 320e23c · 320e23c
1 parent 0389a67
commit 320e23c
Showing 1 changed file with 6 additions and 0 deletions.
diff --git a/main.py b/main.py
@@ -1,8 +1,14 @@
 from fetch_contributions import fetch_github_events, calculate_streaks
 from generate_svg import generate_svg
+import os
 
 if __name__ == '__main__':
     username = 'Vikranth3140'
+
+    # Ensure the output directory exists
+    if not os.path.exists('output'):
+        os.makedirs('output')
+
     events = fetch_github_events(username)
     total_contributions, current_streak, longest_streak = calculate_streaks(events)
     generate_svg(total_contributions, current_streak, longest_streak)