Update app.py

app.py

@@ -376,54 +376,55 @@ def inventory_report(question):
         # Extracting the id for the warehouse with the name "WH"
         warehouse_id = next((item['id'] for item in data['result'] if item['name'] == warehouse_name), None)
         #print(warehouse_id)
-        if warehouse_id is None:
-            print("DB error thrown")
+        if (warehouse_id):
+           
+            # print(f"The id for the warehouse named {name} is: {warehouse_id}")
+            # Step 3: Update the placeholder with the actual warehouse_id
+            for api in apis:
+                if "warehouseId" in api["params"]:
+                    api["params"]["warehouseId"] = warehouse_id
+        
+            print(f"warehouseId: {warehouse_id}")
+            print(f"warehouseId: {apis[1]}")
+           
+            data1 = make_api_request(apis[1]["url"], apis[1]["params"])
+            if (data1):
+                headers = ["S.No", "Warehouse Code", "Warehouse Name", "Customer Code", "Customer Name", "Item Code", "Item Name",
+                           "Currency", "EAN", "UOM", "Quantity", "Gross Weight", "Volume", "Total Value"]
+                table_data = []
+        
+                for index, item in enumerate(data1['result'], start=1):
+                    row = [
+                        index,  # Serial number
+                        item['warehouse']['code'],
+                        item['warehouse']['name'],
+                        item['customer']['code'],
+                        item['customer']['name'],
+                        item['skuMaster']['code'],
+                        item['skuMaster']['name'],
+                        item['currency']['code'],
+                        item['eanUpc'],
+                        item['uom']['code'],
+                        item['totalQty'],
+                        item['grossWeight'],
+                        item['volume'],
+                        item['totalValue']
+                    ]
+                    table_data.append(row)
+        
+                # Convert to pandas DataFrame
+                df = pd.DataFrame(table_data, columns=headers)
+        
+                sdf = SmartDataframe(df, config={"llm": llm_chart})
+        
+                # chart = sdf.chat("Can you draw a bar chart with all avaialble item name and quantity.")
+                chart = sdf.chat(question)
+        
+                return chart
+            else:
+                return "There are no inventory details for the warehouse you have given."
+        else:
             return "Please provide a warehouse name available in the database."
-        # print(f"The id for the warehouse named {name} is: {warehouse_id}")
-        # Step 3: Update the placeholder with the actual warehouse_id
-        for api in apis:
-            if "warehouseId" in api["params"]:
-                api["params"]["warehouseId"] = warehouse_id
-    
-    print(f"warehouseId: {warehouse_id}")
-    print(f"warehouseId: {apis[1]}")
-   
-    data1 = make_api_request(apis[1]["url"], apis[1]["params"])
-    if (data1):
-        headers = ["S.No", "Warehouse Code", "Warehouse Name", "Customer Code", "Customer Name", "Item Code", "Item Name",
-                   "Currency", "EAN", "UOM", "Quantity", "Gross Weight", "Volume", "Total Value"]
-        table_data = []
-
-        for index, item in enumerate(data1['result'], start=1):
-            row = [
-                index,  # Serial number
-                item['warehouse']['code'],
-                item['warehouse']['name'],
-                item['customer']['code'],
-                item['customer']['name'],
-                item['skuMaster']['code'],
-                item['skuMaster']['name'],
-                item['currency']['code'],
-                item['eanUpc'],
-                item['uom']['code'],
-                item['totalQty'],
-                item['grossWeight'],
-                item['volume'],
-                item['totalValue']
-            ]
-            table_data.append(row)
-
-        # Convert to pandas DataFrame
-        df = pd.DataFrame(table_data, columns=headers)
-
-        sdf = SmartDataframe(df, config={"llm": llm_chart})
-
-        # chart = sdf.chat("Can you draw a bar chart with all avaialble item name and quantity.")
-        chart = sdf.chat(question)
-
-        return chart
-    else:
-        return "There are no inventory details for the warehouse you have given."
 
 
 # inventory_report("WH:can you give me a bar chart with item name and quantity for the warehouse WH")