Update app.py

app.py

@@ -24,7 +24,6 @@ generator = pipeline("text2text-generation", model="google/flan-t5-base")
 def classify_comments(category_boxes):
     sentiments = []
     categories = []
-    results = []
     for comment in df['customer_comment']:
         sentiment = classifier(comment)[0]['label']
         category_list = [box for box in category_boxes if box.strip() != '']
@@ -44,29 +43,40 @@ with gr.Blocks() as nps:
             category_list.append(new_category.strip())  # Add new category
         return category_list
 
-    category_boxes = gr.State([])  # Store category input boxes as state
-    
+    def remove_category(category, category_list):
+        category_list.remove(category)  # Remove selected category
+        return category_list
+
     def display_categories(categories):
         category_components = []
         for i, cat in enumerate(categories):
             with gr.Row():
                 gr.Markdown(f"- {cat}")
                 remove_btn = gr.Button("X", elem_id=f"remove_{i}", interactive=True)
-                remove_btn.click(fn=lambda x=cat: remove_category(x, categories), inputs=[], outputs=category_boxes)
+                remove_btn.click(
+                    fn=lambda x=cat: remove_category(x, categories),
+                    inputs=[],
+                    outputs=category_boxes
+                )
             category_components.append(gr.Row())
         return category_components
 
-    def remove_category(category, category_list):
-        category_list.remove(category)  # Remove selected category
-        return category_list
-
+    category_boxes = gr.State([])  # Store category input boxes as state
     category_column = gr.Column()
-    
+
     with gr.Row():
         category_input = gr.Textbox(label="New Category", placeholder="Enter category name")
         add_category_btn = gr.Button("Add Category")
-        add_category_btn.click(fn=add_category, inputs=[category_boxes, category_input], outputs=category_boxes)
-        category_boxes.change(fn=display_categories, inputs=category_boxes, outputs=category_column)
+        add_category_btn.click(
+            fn=add_category,
+            inputs=[category_boxes, category_input],
+            outputs=category_boxes
+        )
+        category_boxes.change(
+            fn=display_categories,
+            inputs=category_boxes,
+            outputs=category_column
+        )
 
     uploaded_file = gr.File(label="Upload CSV", type="filepath")
     template_btn = gr.Button("Use Template")
@@ -79,9 +89,7 @@ with gr.Blocks() as nps:
             file.seek(0)  # Reset file pointer
             import io
             if file.name.endswith('.csv'):
-                file.seek(0)  # Reset file pointer
                 custom_df = pd.read_csv(file, encoding='utf-8')
-                custom_df = pd.read_csv(io.StringIO(content))
             else:
                 return "Error: Uploaded file is not a CSV."
             if 'customer_comment' not in custom_df.columns:
@@ -94,10 +102,12 @@ with gr.Blocks() as nps:
 
     uploaded_file.change(fn=load_data, inputs=uploaded_file, outputs=output)
     template_btn.click(fn=lambda: "Using Template Dataset", outputs=output)
+
     def use_template():
         return ["Product Experience", "Customer Support", "Price of Service", "Other"]
+
     template_btn.click(fn=use_template, outputs=category_boxes)
     category_boxes.change(fn=display_categories, inputs=category_boxes, outputs=category_column)
     classify_btn.click(fn=classify_comments, inputs=category_boxes, outputs=output)
 
-nps.launch()
+nps.launch(share=True)