dataframe fix

app.py

@@ -28,59 +28,51 @@ def predict_emotion(text):
         
         if not text:
             logger.warning("Empty text received")
-            return {
-                "sadness": 0,
-                "joy": 0,
-                "love": 0,
-                "anger": 0,
-                "fear": 0,
-                "surprise": 0
-            }
+            # Return empty DataFrame with correct structure
+            return pd.DataFrame({
+                'emotion': ['sadness', 'joy', 'love', 'anger', 'fear', 'surprise'],
+                'score': [0, 0, 0, 0, 0, 0]
+            })
         
         # Get predictions
         logger.info("Running prediction...")
         predictions = classifier(text)[0]
         logger.info(f"Raw predictions: {predictions}")
         
-        # Convert predictions to a pandas DataFrame for Gradio's BarPlot
+        # Convert to DataFrame and format for visualization
         df = pd.DataFrame(predictions)
+        df.columns = ['emotion', 'score']  # Rename columns
         df = df.sort_values('score', ascending=True)  # Sort for better visualization
         logger.info(f"Processed scores as DataFrame: {df.to_dict()}")
         
-        return (
-            df['label'].tolist(),  # x values (emotions)
-            df['score'].tolist()   # y values (probabilities)
-        )
-        
-    except Exception as e:
-        logger.error(f"Error in prediction: {str(e)}")
-        return {
-            "error": 1.0,
-            "message": f"An error occurred: {str(e)}"
-        }
+        return df
 
 # Create the Gradio interface with error handling
 try:
     logger.info("Setting up Gradio interface...")
     demo = gr.Interface(
         fn=predict_emotion,
-        inputs=gr.Textbox(placeholder="Enter text to analyze...", label="Input Text"),
-        outputs=[
-            gr.BarPlot(
-                title="Emotion Probabilities",
-                x_title="Emotion",
-                y_title="Probability",
-                height=400,
-                vertical=False
-            )
-        ],
-        title="Emotion Detection with DistilBERT",
+        inputs=gr.Textbox(
+            placeholder="Enter text to analyze...", 
+            label="Input Text",
+            lines=4
+        ),
+        outputs=gr.BarPlot(
+            title="Emotion Probabilities",
+            x="emotion",
+            y="score",
+            xlabel="Emotion",
+            ylabel="Probability",
+            height=400
+        ),
+        title="CREATIVE MACHINES: Emotion Detection with DistilBERT",
         description="This app uses the DistilBERT model fine-tuned for emotion detection. Enter any text to analyze its emotional content.",
         examples=[
             ["I am so happy to see you!"],
             ["I'm really angry about what happened."],
             ["The sunset was absolutely beautiful today."],
-            ["I'm worried about the upcoming exam."]
+            ["I'm worried about the upcoming exam."],
+            ["Fear is the mind-killer. I will face my fear."]
         ],
         allow_flagging="never"
     )
@@ -96,4 +88,4 @@ if __name__ == "__main__":
         logger.info("Gradio app launched successfully")
     except Exception as e:
         logger.error(f"Failed to launch Gradio app: {str(e)}")
-        raise
+        raise