update app.py(4)

src/app.py

@@ -65,42 +65,49 @@ if st.session_state.page == "Dashboard":
     st.header("Welcome!!")
     st.text("Here you will find all about your leads.")
     if st.session_state.data_enhancement == True:
+        main_lead_info_file = "data/all_cleaned_companies.json"
+        if os.path.isdir(main_lead_info_file):
+            fig_ind, fig_coun, fig_btype, fig_rev = chart_data.create_chart("data/all_cleaned_companies.json")
+            col1, col2, col3 = st.columns(3)
 
-        fig_ind, fig_coun, fig_btype, fig_rev = chart_data.create_chart("data/all_cleaned_companies.json")
-        col1, col2, col3 = st.columns(3)
-
-        with col1:
-            st.subheader("Industry-wise Distribution")
-            st.plotly_chart(fig_ind, use_container_width=True)
-
-        with col2:
-            st.subheader("Country-wise Distribution")
-            st.plotly_chart(fig_coun, use_container_width=True)
-
-        with col3:
-            st.subheader("Business type-wise Distribution")
-            st.plotly_chart(fig_btype, use_container_width=True)
-        
-        st.subheader("Revenue-based Distribution")
-        st.plotly_chart(fig_rev, use_container_width=True)
-    df_display = chart_data.df_creator_from_json_and_process("data/all_cleaned_companies.json").sort_values(by="score", ascending=False).rename(columns={
-        "company_name": "Company Name",
-        "key_industry": "Industry Type",
-        "industry_type": "Speciality",
-        "street": "Street",
-        "city": "City",
-        "state": "State",
-        "country": "Country",
-        "phone": "Phone",
-        "email": "Email",
-        "company_size": "Number of Employees",
-        "approx_revenue": "Revenue",
-        "business_type": "Business Type",
-        "website_url": "Website",
-        "country": "Country"
-    })
-    st.subheader("All Company Details")
-    st.dataframe(df_display)
+            with col1:
+                st.subheader("Industry-wise Distribution")
+                st.plotly_chart(fig_ind, use_container_width=True)
+
+            with col2:
+                st.subheader("Country-wise Distribution")
+                st.plotly_chart(fig_coun, use_container_width=True)
+
+            with col3:
+                st.subheader("Business type-wise Distribution")
+                st.plotly_chart(fig_btype, use_container_width=True)
+            
+            st.subheader("Revenue-based Distribution")
+            st.plotly_chart(fig_rev, use_container_width=True)
+        else:
+            st.subheader("Do Data Enhancement first in order to view the diagramatic details!!")
+    if os.path.isdir(main_lead_info_file):
+        df_display = chart_data.df_creator_from_json_and_process("data/all_cleaned_companies.json").sort_values(by="score", ascending=False).rename(columns={
+            "company_name": "Company Name",
+            "key_industry": "Industry Type",
+            "industry_type": "Speciality",
+            "street": "Street",
+            "city": "City",
+            "state": "State",
+            "country": "Country",
+            "phone": "Phone",
+            "email": "Email",
+            "company_size": "Number of Employees",
+            "approx_revenue": "Revenue",
+            "business_type": "Business Type",
+            "website_url": "Website",
+            "country": "Country"
+        })
+        st.subheader("All Company Details")
+        st.dataframe(df_display)
+    else:
+        # st.subheader("All Company Details")
+        st.write("_There are no leads yet.Go to Data Enrichment to create leads!!_")
 
 
 if st.session_state.page == "Enrich Companies":