Update app.py

app.py

@@ -1,18 +1,18 @@
 import gradio as gr
-from src.inference import predict_ticket  # Correct import
+from src.inference import predict_ticket  # Now points to the corrected inference.py
 
 def predict_interface(ticket_text):
     result = predict_ticket(ticket_text)
-    issue = result.get('issue_type', 'Unknown')
-    urgency = result.get('urgency_level', 'Unknown')
-    entities = result.get('entities', {})
+    issue = result.get("issue_type", "Unknown")
+    urgency = result.get("urgency_level", "Unknown")
+    entities = result.get("entities", {})
 
-    entities_display = []
-    for key in ['products', 'dates', 'complaints']:
-        values = entities.get(key, [])
-        formatted_values = ', '.join(values) if values else 'None'
-        entities_display.append(f"{key.capitalize()}: {formatted_values}")
-    entities_str = "\n".join(entities_display)
+    # Format entity output
+    lines = []
+    for key in ["products", "dates", "complaints"]:
+        vals = entities.get(key, [])
+        lines.append(f"{key.capitalize()}: {', '.join(vals) if vals else 'None'}")
+    entities_str = "\n".join(lines)
 
     return issue, urgency, entities_str
 
@@ -26,7 +26,7 @@ iface = gr.Interface(
     outputs=[
         gr.Textbox(label="📌 Predicted Issue Type", lines=1),
         gr.Textbox(label="⏱️ Predicted Urgency Level", lines=1),
-        gr.Textbox(label="🧠 Extracted Entities", lines=6)
+        gr.Textbox(label="🧠 Extracted Entities", lines=6),
     ],
     title="📬 Customer Support Ticket Analyzer",
     description=(