Merge branch 'main' of github.com:Fosowl/agenticSeek

requirements.txt

@@ -9,4 +9,6 @@ torch==2.5.1
 ollama==0.4.7
 scipy==1.15.1
 kokoro==0.7.12
-flask==3.1.0
+flask==3.1.0
+soundfile==0.13.1
+termcolor

sources/llm_provider.py

@@ -11,7 +11,7 @@ class Provider:
     def __init__(self, provider_name, model, server_address = "127.0.0.1:5000"):
         self.provider_name = provider_name.lower()
         self.model = model
-        self.server = self.check_address_format(server_address) 
+        self.server = self.check_address_format(server_address)
         self.available_providers = {
             "ollama": self.ollama_fn,
             "server": self.server_fn,
@@ -24,7 +24,7 @@ class Provider:
 
     def check_address_format(self, address):
         """
-        Validate if the address is valid IP. 
+        Validate if the address is valid IP.
         """
         try:
             ip, port = address.rsplit(":", 1)
@@ -34,7 +34,7 @@ class Provider:
         except ValueError as e:
             raise Exception(f"Invalid address format: {e}")
         return address
-    
+
     def respond(self, history, verbose = True):
         """
         Use the choosen provider to generate text.
@@ -60,7 +60,7 @@ class Provider:
         except Exception as e:
             print(f"An error occurred: {e}")
             return False
-    
+
     def server_fn(self, history, verbose = True):
         """
         Use a remote server wit LLM to generate text.
@@ -103,7 +103,7 @@ class Provider:
                 raise Exception("Ollama connection failed. is the server running ?")
             raise e
         return thought
-    
+
     def test_fn(self, history, verbose = True):
         """
         Test function to generate text.
@@ -120,4 +120,4 @@ class Provider:
         echo "Hello world from bash"
         ```
         """
-        return thought
+        return thought

sources/text_to_speech.py

@@ -3,6 +3,8 @@ from IPython.display import display, Audio
 import soundfile as sf
 import subprocess
 import re
+import platform
+
 
 class Speech():
     def __init__(self, language = "english") -> None:
@@ -26,11 +28,16 @@ class Speech():
             speed=1, split_pattern=r'\n+'
         )
         for i, (gs, ps, audio) in enumerate(generator):
-            audio_file = f'sample.wav'
+            audio_file = 'sample.wav'
+            print(audio_file)
             display(Audio(data=audio, rate=24000, autoplay=i==0))
             sf.write(audio_file, audio, 24000) # save each audio file
-            subprocess.call(["afplay", audio_file])
-    
+            if platform.system().lower() != "windows":
+                subprocess.call(["afplay", audio_file])
+            else:
+                import winsound
+                winsound.PlaySound(audio_file, winsound.SND_FILENAME)
+
     def clean_sentence(self, sentence):
         sentence = re.sub(r'`.*?`', '', sentence)
         sentence = re.sub(r'[^a-zA-Z0-9.,!? ]+', '', sentence)

sources/utility.py

@@ -1,20 +1,37 @@
 
 from colorama import Fore
+from termcolor import colored
+import platform
+
 
 def pretty_print(text, color = "info"):
     """
     print text with color
     """
-    color_map = {
-        "success": Fore.GREEN,
-        "failure": Fore.RED,
-        "status": Fore.LIGHTGREEN_EX,
-        "code": Fore.LIGHTBLUE_EX,
-        "warning": Fore.YELLOW,
-        "output": Fore.LIGHTCYAN_EX,
-    }
-    if color not in color_map:
-        print(text)
-        pretty_print("Invalid color in pretty_print", "warning")
-        return
-    print(color_map[color], text, Fore.RESET)
+    if platform.system().lower() != "windows":
+        color_map = {
+            "success": Fore.GREEN,
+            "failure": Fore.RED,
+            "status": Fore.LIGHTGREEN_EX,
+            "code": Fore.LIGHTBLUE_EX,
+            "warning": Fore.YELLOW,
+            "output": Fore.LIGHTCYAN_EX,
+        }
+        if color not in color_map:
+            print(text)
+            pretty_print("Invalid color in pretty_print", "warning")
+            return
+        print(color_map[color], text, Fore.RESET)
+    else:
+        color_map = {
+            "success": "green",
+            "failure": "red",
+            "status": "light_green",
+            "code": "light_blue",
+            "warning": "yello",
+            "output": "cyan",
+            "default": "black"
+        }
+        if color not in color_map:
+            color = "default"
+        print(colored(text, color_map[color]))