src/cli/display.py

@@ -12,7 +12,7 @@ class DisplayFormatter:
         
         sentiment = result["sentiment"]
         confidence = result["confidence"]
-        emoji = "😊" if sentiment == "POSITIVE" else "😞"
+        emoji = "😊" if sentiment == "positive" else "😞"
         
         return f"{emoji} Sentiment: {sentiment}\n📊 Confidence: {confidence:.2%}"
     

src/main.py

@@ -6,10 +6,16 @@ import sys
 from pathlib import Path
 
 # Add parent directory to PYTHONPATH
-sys.path.insert(0, str(Path(__file__).parent.parent))
+sys.path.insert(0, str(Path(__file__).resolve().parent.parent))
 
 from src.cli import InteractiveCLI
-from src.commands import SentimentCommand, FillMaskCommand, TextGenCommand, ModerationCommand, NERCommand
+from src.commands import (
+    FillMaskCommand,
+    ModerationCommand,
+    NERCommand,
+    SentimentCommand,
+    TextGenCommand,
+)
 
 
 def main():
@@ -17,22 +23,27 @@ def main():
     try:
         # Create CLI interface
         cli = InteractiveCLI()
-        
+
         # Register available commands
-        cli.register_command(SentimentCommand())
+        commands_to_register = [
-        cli.register_command(FillMaskCommand())
+            SentimentCommand,
-        cli.register_command(TextGenCommand())
+            FillMaskCommand,
-        cli.register_command(ModerationCommand())
+            TextGenCommand,
-        cli.register_command(NERCommand())
+            ModerationCommand,
-        
+            NERCommand,
+        ]
+        for command in commands_to_register:
+            cli.register_command(command())
+
         # Launch interactive interface
         cli.run()
-        
+
     except KeyboardInterrupt:
         print("\n👋 Stopping program")
     except Exception as e:
         print(f"❌ Error: {e}")
         sys.exit(1)
 
+
 if __name__ == "__main__":
     main()