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,16 +6,10 @@ import sys
 from pathlib import Path
 
 # Add parent directory to PYTHONPATH
-sys.path.insert(0, str(Path(__file__).resolve().parent.parent))
+sys.path.insert(0, str(Path(__file__).parent.parent))
 
 from src.cli import InteractiveCLI
-from src.commands import (
+from src.commands import SentimentCommand, FillMaskCommand, TextGenCommand, ModerationCommand, NERCommand
-    FillMaskCommand,
-    ModerationCommand,
-    NERCommand,
-    SentimentCommand,
-    TextGenCommand,
-)
 
 
 def main():
@@ -23,27 +17,22 @@ def main():
     try:
         # Create CLI interface
         cli = InteractiveCLI()
-
+        
         # Register available commands
-        commands_to_register = [
+        cli.register_command(SentimentCommand())
-            SentimentCommand,
+        cli.register_command(FillMaskCommand())
-            FillMaskCommand,
+        cli.register_command(TextGenCommand())
-            TextGenCommand,
+        cli.register_command(ModerationCommand())
-            ModerationCommand,
+        cli.register_command(NERCommand())
-            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()