Update app.py

app.py

@@ -38,13 +38,17 @@ def fetch_salesforce_data():
         response = sf.query_all(query)
         records = response["records"]
         df = pd.DataFrame(records)
-        df = df.rename(columns={
-            "Product_Name__c": "Product Name",
-            "Modal_Name__c": "Model Name",
-            "Current_Stocks__c": "Current Stocks"
-        })
-        return df[["Product Name", "Model Name", "Current Stocks"]]
+        if not df.empty:
+            df = df.rename(columns={
+                "Product_Name__c": "Product Name",
+                "Modal_Name__c": "Model Name",
+                "Current_Stocks__c": "Current Stocks"
+            })
+            return df[["Product Name", "Model Name", "Current Stocks"]]
+        else:
+            return None
     except Exception as e:
+        print(f"Error fetching Salesforce data: {e}")
         return None
 
 # Function to generate a bar graph from Salesforce data
@@ -62,6 +66,7 @@ def generate_salesforce_bar_chart(df):
         buffer.seek(0)
         return Image.open(buffer)
     except Exception as e:
+        print(f"Error generating bar chart: {e}")
         return None
 
 # Unified function to handle image processing and Salesforce interaction
@@ -117,4 +122,4 @@ def app():
         return interface
 
 if __name__ == "__main__":
-    app().launch(share=True)
+    app().launch(share=True)