fix the pass an index

app.py

@@ -22,7 +22,7 @@ classifier = pipeline(
 logger.info("Pipeline initialized successfully")
 
 # Default DataFrame for BarPlot
-emotion_labels = [pred['label'] for pred in classifier("test")]
+emotion_labels = [pred['label'] for pred in classifier("test input")]
 default_scores = [0] * len(emotion_labels)
 default_df = pd.DataFrame({'label': emotion_labels, 'score': default_scores})
 
@@ -40,10 +40,11 @@ def predict_emotion(text):
         # Get predictions and handle result structure
         logger.info("Running prediction...")
         predictions = classifier(text)
+        logger.debug(f"Raw predictions output: {predictions}")
         
-        # Ensure predictions are in appropriate list-of-dicts format
-        if isinstance(predictions, list) and len(predictions) > 0 and isinstance(predictions[0], dict):
-            df = pd.DataFrame(predictions[0])  # Convert list of dicts to DataFrame
+        # Ensure predictions are a list of dictionaries
+        if isinstance(predictions, list) and len(predictions) > 0 and all(isinstance(item, dict) for item in predictions):
+            df = pd.DataFrame(predictions)  # Convert list of dictionaries to DataFrame
         else:
             logger.error(f"Unexpected predictions format: {predictions}")
             return error_df
@@ -69,7 +70,7 @@ demo = gr.Interface(
         x="label",
         y="score",
         title="Emotion Probabilities",
-        color=["#ff6f61", "#6b5b95", "#88b04b", "#f7cac9", "#92a8d1", "#f7786b"],  # Dynamic color
+        color=["#ff6f61", "#6b5b95", "#88b04b", "#f7cac9", "#92a8d1", "#f7786b"],  # Dynamic colors
         height=400,
         vertical=True
     ),