Update app.py

app.py

@@ -1,151 +1,18 @@
-# import matplotlib.pyplot as plt
-# import nltk
-# import spacy
-# from nltk.sentiment.vader import SentimentIntensityAnalyzer
+
 from transformers import pipeline
 import gradio as gr
 import tensorflow
-# # from reportlab.lib.pagesizes import letter
-# # from reportlab.platypus import SimpleDocTemplate, Image
-
-# nltk.download('vader_lexicon')
-
-# feedbacks = [
-#     "I had a great experience with Inspector Sharma during my visit to the police station. He was very helpful and professional.",
-#     "My experience at the police station was terrible. The officer on duty was rude and uncooperative.",
-#     "I want to express my gratitude to Constable Patel for his prompt response to my complaint. He did an excellent job.",
-#     "I appreciate the efforts of Sub-Inspector Khan in solving my case quickly and efficiently. He demonstrated great problem-solving skills.",
-#     "I had a bad experience at the police station. Officer Gupta was dismissive and didn't take my complaint seriously.",
-#     "I'm thankful to Officer Joshi for helping me with my case. However, the wait time at the police station was excessive, and that needs improvement.",
-#     "I would like to acknowledge the hard work of ACP Singh in maintaining law and order in our area. His dedication is commendable.",
-#     "My recent interaction with the police station was satisfactory. No complaints, but no exceptional service either.",
-#     "I want to express my appreciation to Officer Mehta and Officer Reddy for their quick response to my emergency call. They handled the situation well.",
-#     "I had a positive experience at the police station. Constable Gupta was helpful, but there was some confusion regarding the paperwork that could be streamlined.",
-#     "My interaction with Officer Kapoor and Officer Sharma was neither good nor bad. It was an average experience with no exceptional service.",
-#     "I had a positive experience with Constable Verma. His prompt action was much appreciated.",
-#     "I had an unpleasant experience with Officer Kapoor. He was uncooperative and dismissive of my complaint.",
-#     "Constable Patel provided exceptional service during my visit to the police station. I was impressed with his professionalism.",
-#     "My recent interaction with the police station left much to be desired. The officers seemed disorganized and unresponsive.",
-#     "I had a smooth experience at the police station. No complaints, and the staff was polite and efficient.",
-#     "I appreciate the dedication of Officer Sharma in ensuring the safety of our community. He is an asset to the police force.",
-#     "The service at the police station was disappointing. Officer Gupta was rude and unhelpful.",
-#     "I'm grateful to Sub-Inspector Khan for his assistance in resolving my case. He went above and beyond to help me.",
-#     "Officer Reddy was exceptionally helpful during my recent visit to the police station. I'm thankful for his support.",
-#     "My experience at the police station was unremarkable. The officers did their job, but there was no exceptional service.",
-#     "I had a positive experience with Constable Verma, who handled my case with professionalism and efficiency.",
-#     "I had a negative experience with Officer Joshi. He was unresponsive and didn't take my concerns seriously.",
-#     "The police station needs improvement in terms of service. My recent visit was less than satisfactory.",
-#     "I'm satisfied with the service provided by ACP Singh in our area. He ensures our safety and well-being.",
-#     "I had a disappointing experience at the police station. The officers were uncooperative and unprofessional.",
-#     "I want to acknowledge the hard work of Constable Patel. He is an excellent officer and deserves recognition.",
-#     "I had a mixed experience at the police station. Some aspects were positive, while others need improvement.",
-#     "Officer Mehta provided prompt and efficient assistance during my visit to the police station. I'm thankful for his support.",
-#     "My recent interaction with the police station was average. There were no major issues, but nothing exceptional either."
-# ]
-
-# additional_feedbacks = [
-#     "Officer Reddy and Officer Joshi were both helpful during my recent visit to the police station. They provided the support I needed.",
-#     "I had a positive experience with Officer Mehta and Officer Reddy. Their teamwork was evident in resolving my case.",
-#     "Constable Patel and Officer Gupta were uncooperative during my recent visit. The service was disappointing.",
-#     "Officer Kapoor's behavior was unprofessional during my visit to the police station. I expected better service.",
-#     "I appreciate the assistance of Sub-Inspector Khan and Officer Mehta. They worked together to resolve my complaint.",
-#     "I had a negative experience with Constable Patel. He was dismissive and unhelpful.",
-#     "Officer Sharma and ACP Singh both played crucial roles in maintaining law and order in our area. Their dedication is commendable.",
-#     "My recent interaction with Officer Joshi and Officer Reddy was satisfactory. They handled my concerns efficiently.",
-#     "I had an average experience at the police station. Officer Mehta and Constable Verma provided service without any exceptional aspects.",
-#     "I had a mixed experience with Officer Gupta and Officer Sharma. While they were professional, the process needs improvement.",
-# ]
-# gujarat_feedbacks = [
-#     "I had a positive experience at Gandhinagar Police Station. The officers were helpful and efficient.",
-#     "My visit to Ahmedabad Police Station was satisfactory. The staff handled my complaint professionally.",
-#     "I appreciate the service at Surat Police Station. They resolved my case promptly.",
-#     "Officer Patel at Vadodara Police Station provided exceptional assistance. I'm grateful for their support.",
-#     "My recent interaction with Rajkot Police Station was good. They ensured our safety effectively.",
-#     "I had a mixed experience at Junagadh Police Station. Some aspects were positive, while others need improvement.",
-#     "The police station in Bhavnagar is in need of better service. My recent visit was disappointing.",
-#     "I had a positive experience with Officer Mehta at Anand Police Station. Their prompt action was much appreciated.",
-#     "I want to acknowledge the hard work of the officers at Jamnagar Police Station. Their dedication is commendable.",
-#     "Officer Joshi at Patan Police Station was exceptionally helpful. I'm thankful for their support.",
-#     "I had a negative experience at Porbandar Police Station. The officers were unresponsive and unprofessional.",
-#     "The police station in Nadiad needs improvement. My recent visit was less than satisfactory.",
-#     "I'm satisfied with the service provided by the police in Bharuch, Gujarat. They ensure our safety and well-being.",
-#     "I had an unpleasant experience at the police station in Anand. The officers were dismissive and uncooperative.",
-#     "Officer Sharma at Vadodara Police Station was professional and efficient. I had a positive experience.",
-#     "I had a smooth experience at the police station in Rajkot. No complaints, and the staff was polite and efficient.",
-#     "I appreciate the dedication of the officers at Surat Police Station. They maintain law and order effectively.",
-#     "My experience at the police station in Bhuj was unremarkable. The officers did their job, but there was no exceptional service.",
-#     "I want to express my gratitude to the officers at Bhavnagar Police Station for their prompt response to my complaint.",
-#     "Officer Desai at Porbandar Police Station was helpful and professional during my visit. I had a good experience.",
-#     "My recent interaction with the police in Jamnagar was satisfactory. The officers handled my concerns efficiently.",
-#     "The service at Bhuj Police Station was disappointing. Officer Shah was unhelpful and rude during my visit.",
-#     "I'm thankful to Officer Patel at Gandhinagar Police Station for helping me with my case. Their prompt response was appreciated.",
-#     "My visit to the police station in Anand, Gujarat, was terrible. The officer on duty was dismissive and uncooperative.",
-#     "Officer Pandya at Junagadh Police Station provided excellent assistance. I want to express my gratitude for their support.",
-#     "My recent interaction with the police in Surat was less than satisfactory. The officers seemed disorganized and unresponsive.",
-#     "I had a good experience at Rajkot Police Station. The officers were professional and handled my case effectively.",
-#     "I had an average experience at the police station in Gandhinagar. The service was neither good nor bad.",
-#     "The police station in Bharuch needs improvement. My recent visit was disappointing, and the service was lacking.",
-#     "Officer Mehta at Bhavnagar Police Station was helpful and professional. I appreciate their dedication to their job.",
-#     "I had a mixed experience at Ahmedabad Police Station. Some aspects were positive, while others need improvement.",
-#     "I'm grateful for the service provided by the police in Vadodara, Gujarat. They ensure the safety of our community.",
-# ]
-
-# feedbacks += gujarat_feedbacks
-
 
-# feedbacks += additional_feedbacks
+model_name = "facebook/bart-large-mnli"
 
-issues = ["Misconduct" , "Negligence" , "Discrimination" , "Corruption" , "Violation of Rights" , "Inefficiency" ,
+class_labels = ["Misconduct" , "Negligence" , "Discrimination" , "Corruption" , "Violation of Rights" , "Inefficiency" ,
            "Unprofessional Conduct", "Response Time" , "Use of Firearms" , "Property Damage"]
 
 
-groupByType = TFDebertaForSequenceClassification.from_pretrained(model_name, from_pt=True)
-
-
-
-# pos_count = 0
-# neg_count = 0
-# neutral_count = 0
-
-
-# report = {}
-
-
-
-# ner = spacy.load("en_core_web_sm")
-
-# for feedback in feedbacks:
-#     # location = ""
-#     # doc = ner(feedback)
-#     # for ent in doc.ents:
-#     #     if ent.label_=='GPE':
-#     #         location = ent.text.lower()
-#     #         if location not in report:
-#     #             report[location] = [[0,0,0] , []]
-
-#     # if location:
-        
-#     # score = SentimentIntensityAnalyzer().polarity_scores(feedback)
-#     # sentiment = max(score , key = score.get)
-#     # if sentiment == 'pos':
-#     #     report[location][0][0]+=1
-#     # elif sentiment == 'neg':
-#     #     report[location][0][1]+=1
-#     # else:
-#     #     report[location][0][2]+=1
-
-#     # report[location][1].append(groupByType(feedback , candidate_labels = issues)['labels'][0])
-#     report[feedback] = groupByType(feedback , candidate_labels = issues)['labels'][0]
-
-
-
-
-# # print(report)
-
-
+classifier = pipeline("zero-shot-classification", model=model_name)
 
-def zeroshotclassification(feedback):
-    issue = groupByType(feedback , candidate_labels = issues , multi_label = False)['labels'][0]
+def zeroshotclassification(input_text):
+    issue = classifier(input_text , class_labels , multi_label = False)['labels'][0]
     return issue
 
 iface = gr.Interface(