Update app.py

app.py

@@ -465,8 +465,8 @@ def nlp_pipeline(original_df):
     # return processed_df, budget_cluster_df, problem_cluster_df, location_clusters, problem_clusters
 
     # # Generate project proposals
-    # location_clusters = dict(enumerate(processed_df['Location_Category_Words'].unique()))
-    # problem_clusters = dict(enumerate(processed_df['Problem_Category_Words'].unique()))
+    location_clusters = dict(enumerate(processed_df['Location_Category_Words'].unique()))
+    problem_clusters = dict(enumerate(processed_df['Problem_Category_Words'].unique()))
     project_proposals = create_project_proposals(budget_cluster_df, problem_cluster_df, location_clusters, problem_clusters)
 
     console_messages.append("NLP pipeline completed.")
@@ -507,18 +507,18 @@ def process_excel(file):
 
 
                 
-            # # Ensure location_clusters and problem_clusters are in DataFrame format
-            # if isinstance(location_clusters, pd.DataFrame):
-            #     location_clusters.to_excel(writer, sheet_name='Location_Clusters', index=False)
-            # else:
-            #     console_messages.append("Converting Location Clusters to df")
-            #     pd.DataFrame(location_clusters).to_excel(writer, sheet_name='Location_Clusters', index=False)
+            # Ensure location_clusters and problem_clusters are in DataFrame format
+            if isinstance(location_clusters, pd.DataFrame):
+                location_clusters.to_excel(writer, sheet_name='Location_Clusters', index=False)
+            else:
+                console_messages.append("Converting Location Clusters to df")
+                pd.DataFrame(location_clusters).to_excel(writer, sheet_name='Location_Clusters', index=False)
             
-            # if isinstance(problem_clusters, pd.DataFrame):
-            #     problem_clusters.to_excel(writer, sheet_name='Problem_Clusters', index=False)
-            # else:
-            #     console_messages.append("Converting Problem Clusters to df")
-            #     pd.DataFrame(problem_clusters).to_excel(writer, sheet_name='Problem_Clusters', index=False)
+            if isinstance(problem_clusters, pd.DataFrame):
+                problem_clusters.to_excel(writer, sheet_name='Problem_Clusters', index=False)
+            else:
+                console_messages.append("Converting Problem Clusters to df")
+                pd.DataFrame(problem_clusters).to_excel(writer, sheet_name='Problem_Clusters', index=False)