Update main.py

main.py

@@ -1,38 +1,41 @@
-from math_summarizer import generate_math_summary
-from nlp_summarizer import generate_nlp_summary_and_mindmap
-import openai
-import dotenv
-import os
-
-dotenv.load_dotenv()
-API_KEY = os.getenv('API_KEY')
-
-def create_client(api_key):
-    client = openai.OpenAI(
-        api_key=api_key,
-        base_url="https://glhf.chat/api/openai/v1",
-    )
-    return client
-
-def generate_summary(client, corpus):
-    response = {}
-    print("Generating Math Summary")
-    math_summary = generate_math_summary(corpus)
-    if not math_summary:
-        print("Error generating Math Summary")
-        response['summary_status'] = "error"
-        response['summary'] = None
-        response['mindmap_status'] = "success"
-        response['mindmap'] = None
-        return response
-    else:
-        print("Math Summary Generated Successfully")
-        print("Generating NLP Summary and Mindmap")
-        response = generate_nlp_summary_and_mindmap(client, corpus)
-        print("NLP Summary and Mindmap Generated Successfully")
-        return response
-
-def main(corpus):
-    client = create_client(API_KEY)
-    response = generate_summary(client, corpus)
+from math_summarizer import generate_math_summary
+from nlp_summarizer import generate_nlp_summary_and_mindmap
+import openai
+import dotenv
+import time
+import os
+
+dotenv.load_dotenv()
+API_KEY = os.getenv('API_KEY')
+
+def create_client(api_key):
+    client = openai.OpenAI(
+        api_key=api_key,
+        base_url="https://glhf.chat/api/openai/v1",
+    )
+    return client
+
+def generate_summary(client, corpus):
+    response = {}
+    print("Generating Math Summary")
+    math_summary = generate_math_summary(corpus)
+    if not math_summary:
+        print("Error generating Math Summary")
+        response['summary_status'] = "error"
+        response['summary'] = None
+        response['mindmap_status'] = "success"
+        response['mindmap'] = None
+        return response
+    else:
+        print("Math Summary Generated Successfully")
+        print("Generating NLP Summary and Mindmap")
+        response = generate_nlp_summary_and_mindmap(client, corpus)
+        print("NLP Summary and Mindmap Generated Successfully")
+        return response
+
+def main(corpus):
+    start_time = time.time()
+    client = create_client(API_KEY)
+    response = generate_summary(client, corpus)
+    print(f"Total timetaken: {time.time() - start_time} seconds")
     return response