Merge pull request 'develop' (#1) from develop into main

Reviewed-on: Cyril/AI-Lab-Transformers-CLI-Playground#1
This commit is contained in:
Cyril 2025-10-12 09:41:41 +00:00
commit 71c2c2c3da
2 changed files with 22 additions and 11 deletions

View File

@ -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%}"

View File

@ -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():
@ -19,11 +25,15 @@ def main():
cli = InteractiveCLI()
# Register available commands
cli.register_command(SentimentCommand())
cli.register_command(FillMaskCommand())
cli.register_command(TextGenCommand())
cli.register_command(ModerationCommand())
cli.register_command(NERCommand())
commands_to_register = [
SentimentCommand,
FillMaskCommand,
TextGenCommand,
ModerationCommand,
NERCommand,
]
for command in commands_to_register:
cli.register_command(command())
# Launch interactive interface
cli.run()
@ -34,5 +44,6 @@ def main():
print(f"❌ Error: {e}")
sys.exit(1)
if __name__ == "__main__":
main()