Update app.py

app.py

@@ -11,31 +11,22 @@ import plotly.express as px
 from PIL import Image
 
 categories_keywords = {
-    "Application Status": ["application status", "application", "status", "submitted", "processing", "pending", "approval", "rejected", "accepted"],
-    "Volunteering": ["volunteer", "volunteering", "help out", "assist", "volunteer work", "volunteer opportunities"],
-    "Certificates": ["certificate", "certificates", "completion", "certification", "accreditation", "proof", "document", "certified"],
-    "Job Opportunities": ["job", "opportunity", "career", "vacancy", "position", "employment", "hiring", "recruitment", "internship"],
-    "Surveys and Forms": ["survey", "form", "forms", "questionnaire", "feedback form", "response", "fill out", "submission"],
-    "General Queries": ["hello", "hi", "help", "general", "query", "question", "info", "information", "inquiry", "ask"],
-    "Spam": ["spam", "subscribe", "remove", "stop", "junk", "block", "opt-out", "watch my video", "click on this link", "win prize"],
-    "Rescheduling and Postponing": ["reschedule", "postpone", "delay", "change date", "new time", "rearrange", "shift", "adjust timing"],
-    "Contact and Communication Issues": ["contact", "communicate", "communication", "reach out", "phone", "email", "address", "details"],
-    "Email and Credentials Issues": ["email", "credentials", "login", "password", "access", "username", "account", "verification", "reset"],
-    "Timing and Scheduling": ["timing", "schedule", "scheduling", "time", "appointment", "availability", "calendar", "book", "slot"],
-    "Salary and Benefits": ["salary", "benefits", "pay", "compensation", "wages", "earnings", "package", "remuneration", "incentives"],
-    "Technical Issues": ["technical", "issue", "problem", "error", "bug", "glitch", "fix", "troubleshoot", "support"],
-    "End of Conversation": ["bye", "thank you", "thanks", "goodbye", "see you", "later", "end", "close", "sign off"],
-    "Start of Conversation": ["start", "begin", "hello", "hi", "initiate", "greet", "greeting", "open", "commence"],
-    "Feedback": ["feedback", "comments", "review", "opinion", "suggestion", "critique", "rating"],
-    "Online Meetings": ["meeting", "meeting code", "passcode", "join meeting", "zoom", "zoom link"],
-    "Event Inquiries": ["event", "webinar", "meeting", "conference", "session", "seminar", "workshop", "invitation"],
-    "Payment Issues": ["payment", "billing", "transaction", "charge", "fee", "invoice", "refund", "receipt"],
-    "Registration Issues": ["registration", "register", "sign up", "enroll", "join", "signup", "enrollment"],
-    "Service Requests": ["service", "support", "request", "assistance", "help", "aid", "maintenance"],
-    "Account Issues": ["account", "profile", "update", "activation", "deactivation", "credentials", "reset"],
-    "Product Information": ["product", "service", "details", "info", "information", "specifications", "features"],
-    "Order Status": ["order", "status", "tracking", "shipment", "delivery", "purchase", "dispatch"],
-    "Miscellaneous": []
+    'Application Status': ['application', 'applied', 'update on my application', 'result of my application', 'selected', 'selection process', 'apply', 'fellow', 'lesson plan', 'status of my application', 'application update', 'application status', 'applied for'],
+    'Volunteering': ['volunteering', 'volunteer', 'volunteering certificate', 'resume my volunteering', 'volunteering journey', 'volunteering with TFI', 'volunteering opportunities', 'volunteer work', 'volunteer program'],
+    'Certificates': ['certificate', 'certificates', 'certificate of completion', 'volunteer certificate', 'issue certificate'],
+    'Job Opportunities': ['job', 'vacancy', 'Talent Acquisition Executive job', 'opportunity', 'job opening', 'job position', 'career opportunities'],
+    'Surveys and Forms': ['survey', 'form', 'fill out the survey', 'application form', 'survey link', 'survey form', 'form submission'],
+    'General Queries': ['query', 'queries', 'questions', 'feedback', 'loved', 'overwhelming', 'general question', 'inquiry', 'query about'],
+    'Spam': ['free recharge', 'offer', 'click the link', 'https'],
+    'Rescheduling and Postponing': ['reschedule', 'postpone', 'cancellation', 'date', 'time slot', 'change date', 'change time', 'reschedule appointment'],
+    'Contact and Communication Issues': ['call', 'phone', 'contact', 'not received', 'contact support', 'phone call', 'call back', 'internet'],
+    'Email and Credentials Issues': ['email', 'credentials', 'received', 'email issue', 'email problem', 'credential issue', 'login problem'],
+    'Timing and Scheduling': ['session', 'time', 'interview', 'baje', 'schedule time', 'meeting time', 'appointment time'],
+    'Salary and Benefits': ['salary', 'increment', 'accommodation', 'training period', 'reside', 'stipend', 'pay', 'wage', 'salary details', 'benefits information'],
+    'Technical Issues': ['network issues', 'zoom meeting', 'passcode', 'technical', 'issue','technical problem', 'system issue', 'technical support'],
+    'Complaint Handling': ['help', 'i need help', 'Help me', 'complaint', 'issue is unresolved', 'unsatisfied', 'bad experience'],
+    'End of Conversation': ['thanks', 'thankss', 'thank u', 'thank you', 'ok', 'okay', 'done', 'joining', 'sounds good', 'goodbye', 'end chat', 'end'],
+    'Miscellaneous': []
 }
 
 
@@ -169,43 +160,41 @@ def generate_bar_chart(df, num_clusters_to_display):
     img = Image.open(buf)
     return img
 
-def main(file, num_categories_to_display):
+def main(file, num_clusters_to_display):
     try:
         df = pd.read_csv(file)
         
+        # Filter by 'Fallback Message shown'
+        df = df[df['Answer'] == 'Fallback Message shown']
         
         df = preprocess_data(df)
-
-        df = df[df['Category'] != 'Miscellaneous']
         
-        # Get category sizes and sort by size in descending order
+        # Get category sizes and sort by size in ascending order
         category_sizes = df['Category'].value_counts().reset_index()
         category_sizes.columns = ['Category', 'Count']
-        sorted_categories = category_sizes.sort_values(by='Count', ascending=False)['Category'].tolist()
-        sorted_categories_sm = category_sizes.sort_values(by='Count', ascending=True)['Category'].tolist()
-
+        sorted_categories = category_sizes.sort_values(by='Count', ascending=True)['Category'].tolist()
         
-        # Get the largest x categories as specified by num_categories_to_display
-        largest_categories = sorted_categories[:num_categories_to_display]
-        smallest_categories = sorted_categories_sm[:num_categories_to_display]
+        # Get the largest x categories as specified by num_clusters_to_display
+        largest_categories = sorted_categories[:num_clusters_to_display]
         
         # Filter the dataframe to include only the largest categories
         filtered_df = df[df['Category'].isin(largest_categories)]
-        filtered_cloud_df = df[df['Category'].isin(smallest_categories)]
         
         # Sort the dataframe by Category
         filtered_df = filtered_df.sort_values(by='Category')
-        filtered_cloud_df = filtered_cloud_df.sort_values(by='Category')
         
-        wordcloud_img = generate_wordcloud(filtered_cloud_df)  # Word cloud based on the entire dataframe
-        bar_chart_img = generate_bar_chart(df, num_categories_to_display)  # Bar chart based on the filtered dataframe
+        wordcloud_img = generate_wordcloud(filtered_df)
+        bar_chart_img = generate_bar_chart(filtered_df, num_clusters_to_display)
 
         with tempfile.NamedTemporaryFile(delete=False, suffix=".csv") as tmpfile:
             filtered_df.to_csv(tmpfile.name, index=False)
+            csv_file_path = tmpfile.name
+
+        return csv_file_path, wordcloud_img, bar_chart_img
+    except Exception as e:
         print(f"Error: {e}")
         return str(e), None, None
 
-
 interface = gr.Interface(
     fn=main,
     inputs=[
@@ -222,4 +211,3 @@ interface = gr.Interface(
 )
 
 interface.launch(share=True)
-