upgraded

api/__init__.py

@@ -756,7 +756,7 @@ class LoLLMsAPPI():
                     anti_prompt_to_remove = prompt.lower()
                     
             if not detected_anti_prompt:
-                    ASCIIColors.green(f"generated:{len(self.current_generated_text)} words", end='\r')
+                    ASCIIColors.green(f"generated:{len(self.current_generated_text.split())} words", end='\r', flush=True)
                     self.socketio.emit('message', {
                                                     'data': self.current_generated_text, 
                                                     'user_message_id':self.current_user_message_id, 

api/extension.py

@@ -1,18 +0,0 @@
-# File : extension.py
-# Author : ParisNeo
-# Description :
-# This is the main class to be imported by the extension
-# it gives your code access to the model, the callback functions, the model conditionning etc
-from config import load_config, save_config
-
-__author__ = "parisneo"
-__github__ = "https://github.com/ParisNeo/lollms-webui"
-__copyright__ = "Copyright 2023, "
-__license__ = "Apache 2.0"
-
-class Extension():
-    def __init__(self, metadata_file_path:str, app) -> None:
-        self.app = app
-        self.metadata_file_path = metadata_file_path
-        self.config = load_config(metadata_file_path)
-

c_webui.bat

@@ -43,7 +43,7 @@ echo Testing internet connection
 ping -n 1 google.com >nul
 if %errorlevel% equ 0 (
     echo Internet Connection working fine
-    
+
     REM Install Git
     echo Checking for Git...
     where git >nul 2>nul

requirements.txt

@@ -7,8 +7,9 @@ pyyaml
 markdown
 gevent
 gevent-websocket
-lollms
+lollms==2.1.14
 langchain
 requests
 eventlet
-websocket-client
+websocket-client
+pywhispercpp

webui.bat

@@ -58,7 +58,6 @@ goto END
 :INTERNET_OK
 echo \e[32mInternet connection working fine
 
-
 REM Check if Git is installed
 echo "Checking for git..."
 where git >nul 2>&1