Add descriptions to AI models

src/providers/aihorde.py

@@ -13,7 +13,7 @@ class AIHordeProvider(BaseProvider):
     STATUS_URL = "https://stablehorde.net/api/v2/generate/text/status/"
     API_KEY = "0000000000"
     model = "PygmalionAI/pygmalion-7b"
-    
+    description = "AI Horde is a crowdsourced distributed cluster of Image generation workers and text generation workers."
 
     def ask(self, prompt, chat, **kwargs):
         self.API_KEY = self.data.get("api_key", "0000000000")

src/providers/analogdiffusion.py

@@ -3,3 +3,4 @@ from .basehfimage import BaseHFImageProvider
 class AnalogDiffusionProvider(BaseHFImageProvider):
     name = "Analog Diffusion"
     provider = "wavymulder/Analog-Diffusion"
+    description = "Analog Diffusion is a model that can generate images from a prompt."

src/providers/nitrodiffusion.py

@@ -2,4 +2,5 @@ from .basehfimage import BaseHFImageProvider
 
 class NitroDiffusionProvider(BaseHFImageProvider):
     name = "Nitro Diffusion"
-    provider = "nitrosocke/Nitro-Diffusion"
+    provider = "nitrosocke/Nitro-Diffusion"
+    description = "Nitro Diffusion is a model that can generate images from a prompt."

src/providers/openaiimage.py

@@ -8,7 +8,7 @@ import json
 from gi.repository import Gtk, Adw, GLib
 
 
-class BaseOpenAIImageProvider(BaseProvider):
+class BaseOpenAIImageProvider(BaseImageProvider):
     model = None
     api_key_title = "API Key"
 
@@ -33,9 +33,9 @@ class BaseOpenAIImageProvider(BaseProvider):
         if self.model:
             prompt = self.chunk(prompt)
             try:
-                response = client.images.generate(
+                response = self.client.images.generate(
                     model=self.model,
-                    prompt=self.prompt,
+                    prompt=prompt,
                     size="1024x1024",
                     quality="standard",
                     n=1,
@@ -110,7 +110,9 @@ class BaseOpenAIImageProvider(BaseProvider):
 class DallE2(BaseOpenAIImageProvider):
     name = "DALL·E 2"
     model = "dall-e-2"
+    description = "DALL·E is a AI system that can create realistic images and art from a description in natural language. "
 
 class DallE3(BaseOpenAIImageProvider):
     name = "DALL·E 3"
-    model = "dall-e-3"
+    model = "dall-e-3"
+    description = "DALL·E is a AI system that can create realistic images and art from a description in natural language. "

src/providers/openassistantsft1pythia12b.py

@@ -3,6 +3,7 @@ from .hfbasechat import BaseHFChatProvider
 class HuggingFaceOpenAssistantSFT1PythiaProvider(BaseHFChatProvider):
     name = "Open-Assistant SFT-1 12B"
     provider = "OpenAssistant/oasst-sft-4-pythia-12b-epoch-3.5"
+    description = "OpenAssistant's SFT-1 Pythia 12B model"
 
     def make_prompt(self, prompt, chat):
         p = ""

src/providers/openjourney.py

@@ -2,4 +2,5 @@ from .basehfimage import BaseHFImageProvider
 
 class OpenJourneyProvider(BaseHFImageProvider):
     name = "Open Journey"
-    provider = "prompthero/openjourney-v4"
+    provider = "prompthero/openjourney-v4"
+    description = "Open Journey is a model that can generate images from a prompt."

src/providers/portraitplus.py

@@ -2,4 +2,5 @@ from .basehfimage import BaseHFImageProvider
 
 class PortraitPlusProvider(BaseHFImageProvider):
     name = "Portrait Plus"
-    model = "wavymulder/portraitplus"
+    model = "wavymulder/portraitplus"
+    description = "Portrait Plus is a model that can generate images from a prompt."

src/providers/provider_item.py

@@ -28,7 +28,7 @@ class Provider(Adw.ExpanderRow):
         self.provider_type.set_label(self.provider.provider_type.value)
         match self.provider.provider_type:
             case ProviderType.IMAGE:
-                self.provider_type.add_css_class("badge-silver")
+                self.provider_type.add_css_class("badge-titanium")
             case ProviderType.CHAT:
                 self.provider_type.add_css_class("badge-gold")
             case ProviderType.VOICE:
@@ -36,7 +36,7 @@ class Provider(Adw.ExpanderRow):
             case ProviderType.TEXT:
                 self.provider_type.add_css_class("badge-tin")
             case ProviderType.MOVIE:
-                self.provider_type.add_css_class("badge-titanium")
+                self.provider_type.add_css_class("badge-silver")
 
         self.enable_switch.set_active( self.app.data["providers"][self.provider.slug]["enabled"])
 

src/providers/robertasquad2.py

@@ -3,6 +3,7 @@ from .hfbasechat import BaseHFChatProvider
 class RobertaSquad2Provider(BaseHFChatProvider):
     name = "Roberta Squad2"
     provider = "deepset/roberta-base-squad2"
+    description = "A model for Question Answering on SQuAD2"
 
     def make_prompt(self, prompt, chat):
         context = ""

src/providers/stablediffusion.py

@@ -2,4 +2,5 @@ from .basehfimage import BaseHFImageProvider
 
 class StableDiffusionProvider(BaseHFImageProvider):
     name = "Stable Diffusion"
-    provider = "stabilityai/stable-diffusion-2-1"
+    provider = "stabilityai/stable-diffusion-2-1"
+    description = "Stable Diffusion is a model that can generate images from a prompt."