Update app.py

app.py

@@ -40,9 +40,9 @@ def process_file(file):
     is_contract_flag, classification = is_contract(text)
     if is_contract_flag:
         parties = extract_parties(text)
-        return "✅ This is a contract.", parties
+        return "✅ This is a contract.", ", ".join(parties)
     else:
-        return "❌ This is NOT a contract.", []
+        return "❌ This is NOT a contract.", ""
 
 # Gradio interface
 iface = gr.Interface(
@@ -50,7 +50,7 @@ iface = gr.Interface(
     inputs=gr.File(file_types=[".txt", ".pdf", ".docx"], label="Upload a document"),
     outputs=[
         gr.Textbox(label="Classification Result"),
-        gr.Label(label="Detected Parties")
+        gr.Textbox(label="Detected Parties (ORG/PER)")
     ],
     title="Contract Classifier with RoBERTa",
     description="Upload a document (.pdf, .txt, .docx) to detect if it's a contract and extract involved parties using RoBERTa."