Add sample output flag and update sample descriptions in API response

app.py

@@ -4,6 +4,7 @@ import requests
 import json
 import time
 from datetime import datetime
+import time
 import pandas as pd
 from fastapi import FastAPI
 from fastapi.middleware.cors import CORSMiddleware
@@ -51,6 +52,7 @@ class PredictionInput(BaseModel):
     levelOfSugar: str
     listPricePerUnitMl: float
     weightPerUnitMl: float
+    sampleOutput: bool
 
 class inputtext(BaseModel):
     inputtext:str
@@ -112,7 +114,8 @@ def run_pred_pipeline(input: PredictionInput):
     print(f"Running the pipeline : {datetime.now().strftime('%Y-%m-%d %H:%M:%S')} ")
 
     ## Hardcoding for testing purposes ##
-    if input.dict().get('sampleOutput') == 'true':
+    if input.dict().get('sampleOutput') == True:
+        time.sleep(4)
         temp_predictions_dict = generate_random_predictions()
         sample_sim_attr = get_sample_similarity_attr()
         data_out = {

helper.py

@@ -36,7 +36,7 @@ def process_api_response(json_response):
 def get_sample_similarity_attr():
     sample_sim = {
         "GB1263": {
-            "description": "THIS IS A SAMPLE DESCRIPTION",
+            "description": "MILKA 270G MILK",
             "sell_in_volume": 44426.88,
             "similarity": 66.67,
             "distribution": [
@@ -77,7 +77,7 @@ def get_sample_similarity_attr():
             ]
         },
         "GB1264": {
-            "description": "THIS IS A SAMPLE DESCRIPTION",
+            "description": "MILKA 270G WN",
             "sell_in_volume": 26840.97,
             "similarity": 66.67,
             "distribution": [
@@ -118,7 +118,7 @@ def get_sample_similarity_attr():
             ]
         },
         "GB1450": {
-            "description": "THIS IS A SAMPLE DESCRIPTION",
+            "description": "MILKA LARGE BLOCK OREO",
             "sell_in_volume": 32346.0,
             "similarity": 66.67,
             "distribution": [
@@ -159,7 +159,7 @@ def get_sample_similarity_attr():
             ]
         },
         "GB1561": {
-            "description": "THIS IS A SAMPLE DESCRIPTION",
+            "description": "MIXED DUO PREFILL",
             "sell_in_volume": 3265.92,
             "similarity": 77.78,
             "distribution": [
@@ -180,7 +180,7 @@ def get_sample_similarity_attr():
             ]
         },
         "GB1650": {
-            "description": "THIS IS A SAMPLE DESCRIPTION",
+            "description": "CAD TWIRL MINT",
             "sell_in_volume": 0.0,
             "similarity": 77.78,
             "distribution": [