Update app.py

app.py

@@ -23,6 +23,28 @@ def load_model():
 # Load model at startup
 load_model()
 
+# Define formality range responses
+FORMALITY_RESPONSES = {
+    "very_informal": "Your speech formality score is {}/100 — quite informal! Your tone is very casual, which could be great for personal conversations with friends or informal settings. However, in professional or formal situations, you may want to avoid using overly relaxed language. Consider using more respectful and polished language when addressing colleagues, clients, or superiors in business settings.",
+    "informal": "Your speech formality score is {}/100 — a bit informal but showing improvement! While your tone is still casual, it's better suited for friendly exchanges or informal conversations. However, for interviews, meetings, or any formal discussion, it's a good idea to increase the level of professionalism in your language. Using more polite phrases and professional vocabulary can enhance your communication in these contexts.",
+    "balanced": "Your speech formality score is {}/100 — a balanced mix of casual and formal! You're in a comfortable middle ground, which works well for most social interactions and some professional settings. It's appropriate for emails or casual discussions with colleagues. However, if you're presenting to a high-level audience or in formal meetings, consider polishing your language slightly by avoiding colloquialisms and being more direct in your communication.",
+    "formal": "Your speech formality score is {}/100 — mostly formal with a hint of informality! This tone is well-suited for professional emails, presentations, and discussions in business meetings. Your language shows respect and professionalism while still remaining approachable. For more formal settings like conferences or addressing senior executives, you may want to avoid casual phrases and adopt more sophisticated language to leave a stronger impression.",
+    "very_formal": "Your speech formality score is {}/100 — highly formal and polished! Your tone is perfect for formal meetings, academic presentations, or professional interactions where respect and authority are key. It's appropriate for official correspondence, addressing clients or superiors, and speaking at conferences. Keep in mind that, in more casual settings or among peers, this level of formality might feel a bit too stiff. Tailor your approach depending on your audience, but you're definitely on track for professional success."
+}
+
+def get_formality_category(formal_percent):
+    """Determine formality category based on percentage"""
+    if formal_percent <= 20:
+        return "very_informal"
+    elif formal_percent <= 40:
+        return "informal"
+    elif formal_percent <= 60:
+        return "balanced"
+    elif formal_percent <= 80:
+        return "formal"
+    else:
+        return "very_formal"
+
 @app.route("/", methods=["GET"])
 def home():
     return jsonify({"message": "Formality Classifier API is running! Use /predict to classify text."})
@@ -49,12 +71,16 @@ def predict_formality():
         # Convert score to formality classification
         formal_percent = round(score * 100)
         informal_percent = 100 - formal_percent
+        
+        # Get the appropriate formality category and response
+        category = get_formality_category(formal_percent)
+        detailed_response = FORMALITY_RESPONSES[category].format(formal_percent)
 
         return jsonify({
             "formality_score": round(score, 3),
             "formal_percent": formal_percent,
             "informal_percent": informal_percent,
-            "classification": f"Your speech is {formal_percent}% formal and {informal_percent}% informal."
+            "classification": detailed_response
         })
 
     except Exception as e: