Update app.py

app.py

@@ -77,11 +77,11 @@ def detect_context(input_text, threshold=0.03):
     label_probabilities = list(zip(labels, probabilities))
 
     # Filter contexts with confidence >= threshold
-    high_confidence_contexts = [label in label_probabilities if score >= threshold]
+    high_confidence_contexts = [[(label, score) for label, scores in label_probabilities if score >= threshold]
 
     # If no contexts meet the threshold, default to "general"
     if not high_confidence_contexts:
-        high_confidence_contexts = [("general")]
+        high_confidence_contexts = [("general", 1.0)]
 
     return high_confidence_contexts
 
@@ -109,9 +109,7 @@ def process_request(input_text):
 def gradio_interface(input_text):
     translation, contexts = process_request(input_text)
     # Format the output
-    output = f"Translation: {translation}\n\nContexts:\n"
-    for context in contexts:
-        output += f"- {context}\n"
+    output = f"Translation: {translation}\n"
     return output.strip()
 
 # Create the Gradio interface