Update app.py

app.py

@@ -86,20 +86,32 @@ def launch(text_input):
     decoded_labels = label_encoder.inverse_transform(predicted_classes)
 
     map_data = pd.read_excel("ISCO-08 EN Structure and definitions.xlsx")
-    isco_description = map_data[map_data['ISCO 08 Code']==decoded_labels]['Title EN']
-
-    print("Most likely ISCO code is {} and probability is {}".format(decoded_labels,highest_probabilities))
-
+    # Retrieve the ISCO description based on the decoded label
+    isco_description = map_data[map_data['ISCO 08 Code'] == decoded_labels[0]]['Title EN'].values
+    
+    # Print for debugging (optional)
+    print(f"Most likely ISCO code is {decoded_labels[0]} and probability is {highest_probabilities[0]}")
+    
     # Create descriptive text for the output
-    #result_text = "Most likely ISCO code is {} and probability is {:.2f}".format(decoded_labels[0], highest_probabilities[0])
-    result_text = f"Predicted ISCO Code: {decoded_labels[0]}\nProbability: {highest_probabilities[0]:.2f}\nISCO Description:{isco_description}"
-    return result_text #[decoded_labels[0], highest_probabilities[0]]
-
+    result_text = (
+        f"Predicted ISCO Code: {decoded_labels[0]}\n"
+        f"Probability: {highest_probabilities[0]:.2f}\n"
+        f"ISCO Description: {isco_description[0] if len(isco_description) > 0 else 'Description not found'}"
+    )
+    
+    return result_text
 
+# Define the Gradio interface
 iface = gr.Interface(
     fn=launch,
-    inputs=gr.Textbox(lines=2, placeholder="Enter job title in any language (e.g., Software Engineer) and description here (e.g., Develops and maintains software applications)..."),
-    outputs=gr.Textbox(lines=2, placeholder="Predicted ISCO Code: <result> Probability: <result> ISCO Description: <result>")
+    inputs=gr.Textbox(
+        lines=2, 
+        placeholder="Enter job title in any language (e.g., Software Engineer) and description here (e.g., Develops and maintains software applications)..."
+    ),
+    outputs=gr.Textbox(
+        lines=4, 
+        placeholder="Predicted ISCO Code: <result>\nProbability: <result>\nISCO Description: <result>"
+    )
 )
 
 iface.launch()