Overhaul

app.py

@@ -1,5 +1,6 @@
 from fastapi import FastAPI
 from fastapi.middleware.cors import CORSMiddleware
+from fastapi.middleware.cors import CORSMiddleware
 
 # Define the FastAPI app
 app = FastAPI(docs_url="/")
@@ -191,6 +192,8 @@ def search(
         descriptions = []
         images = []
 
+        # Set the OpenAI API key
+        openai.api_key = "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"
         # Set the OpenAI API key
         openai.api_key = "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"
 
@@ -455,30 +458,26 @@ def search(
     runtime = f"{end_time - start_time:.2f} seconds"
 
     # Create a list of dictionaries to store the results
-    results = [
-        {
-            "id": i,
-            "title": titles[i],
-            "author": authors[i],
-            "publisher": publishers[i],
-            "image_link": images[i],
-            "labels": classes[i]["labels"][0:2],
-            "label_confidences": classes[i]["scores"][0:2],
-            "summary": summaries[i][0]["summary_text"],
-            "similar_books": similar_books[i]["sorted_by_similarity"],
-            "checkpoints": [
-                {
-                    "Google Books Time": first_checkpoint_time,
-                    "OpenAlex Time": second_checkpoint_time,
-                    "OpenAI Time": third_checkpoint_time,
-                    "Classification Time": classification_time,
-                    "Summarization Time": summarization_time,
-                    "Similarity Computing Time": similarity_time,
-                }
-            ],
-            "total_runtime": runtime,
-        }
-        for i in range(len(combined_data))
-    ]
+    results = []
+    for i in range(len(titles)):
+        results.append(
+            {
+                "id": i,
+                "title": titles[i],
+                "author": authors[i],
+                "publisher": publishers[i],
+                "image_link": images[i],
+                "labels": classes[i]["labels"][0:2],
+                "label_confidences": classes[i]["scores"][0:2],
+                "summary": summaries[i][0]["summary_text"],
+                "similar_books": similar_books[i]["sorted_by_similarity"],
+                "runtime": {
+                    "total": runtime,
+                    "classification": classification_time,
+                    "summarization": summarization_time,
+                    "similarity": similarity_time,
+                },
+            }
+        )
 
     return results