Update app.py

app.py

@@ -30,15 +30,22 @@ def spam_detection(input_text):
 
 def negative_zero_shot(input_text):
     try:
-        return model.generate_content(f'Issues should be from ["Misconduct" , "Negligence" , "Discrimination" , "Corruption" , "Violation of Rights" , "Inefficiency" , "Unprofessional Conduct", "Response Time" , "Use of Firearms" , "Property Damage"] only. Give me the issue faced by the feedback giver in less than four words: {input_text}').text
+        return model.generate_content(f'Issues should be from ["Misconduct" , "Negligence" , "Discrimination" , "Corruption" , "Violation of Rights" , "Inefficiency" , "Unprofessional Conduct", "Response Time" , "Use of Firearms" , "Property Damage"] only. Give me the issue faced by the feedback giver in less than four words. If no specific category is detected, take "Offensive" as default. Feedback: {input_text}').text
     except Exception as e:
         return "Offensive"
 
 def positive_zero_shot(input_text):
     try:
-        return model.generate_content(f'Issues should be from ["Miscellaneous", "Tech-Savvy Staff" , "Co-operative Staff" , "Well-Maintained Premises" , "Responsive Staff"] only. Give me the issue faced by the feedback giver in less than four words: {input_text}').text
+        return model.generate_content(f'Issues should be from ["Miscellaneous", "Tech-Savvy Staff" , "Co-operative Staff" , "Well-Maintained Premises" , "Responsive Staff"] only. Give me the issue faced by the feedback giver in less than four words. If no specific category is detected, take "Appreciation" as default. Feedback: {input_text}').text
     except Exception as e:
-        return "Offensive"
+        return "Appreciation"
+        
+def which_department(input_text):
+    try:
+        return model.generate_content(f'Departments should be from ["Crime branch", "Rajasthan Armed Constabulary (RAC)", "State Special Branch", "Anti Terrorist Squad (ATS)", "Planning and Welfare", "Training", "Forensic Science laboratory", "Telecommunications", "Cybersecurity", "Traffic Police"] only. Give me the department about which the user is giving feedback. If no specific department is mentioned, take "Crime Branch" as default. Feedback: {input_text}').text
+    except Exception as e:
+        return "Crime branch"
+
 def pipeline(input_text):
 
     input_text = translate(input_text)
@@ -46,25 +53,26 @@ def pipeline(input_text):
     if spam_detection(input_text):
 
         sent = float(sentiment(input_text))
+        dept = which_department(input_text)
         
         if sent > 0:
             
-            return str(sent), positive_zero_shot(input_text)
+            return str(sent), positive_zero_shot(input_text), dept
         
         elif sent < 0:
             
-            return str(sent),  negative_zero_shot(input_text)
+            return str(sent),  negative_zero_shot(input_text), dept
         
         else:
             
-            return "0", "No issue"
+            return "0", "No issue", dept
     else:
-        return "42", "Spam"
+        return "42", "Spam", "No department"
 
 iface = gr.Interface(
     fn = pipeline,
     inputs = ["text"],
-    outputs = ["text", "text"]
+    outputs = ["text", "text", "text"]
 )
 
-iface.launch(share=True)
+iface.launch()