syntaxmatrix 2.5.5.5__py3-none-any.whl → 2.6.2__py3-none-any.whl

syntaxmatrix/core.py

@@ -16,7 +16,6 @@ from .file_processor import process_admin_pdf_files
 from google.genai import types
 from .vector_db import query_embeddings
 from .vectorizer import embed_text
-from syntaxmatrix.settings.prompts import SMXAI_CHAT_ID, SMXAI_CHAT_INSTRUCTIONS, SMXAI_WEBSITE_DESCRIPTION
 from typing import List, Generator
 from .auth import init_auth_db
 from . import profiles as _prof
@@ -28,20 +27,26 @@ from html import unescape
 from .plottings import render_plotly, pyplot, describe_plotly, describe_matplotlib
 from threading import RLock
 from syntaxmatrix.settings.model_map import GPT_MODELS_LATEST
-
+from syntaxmatrix.settings.prompts import(
+    SMXAI_CHAT_IDENTITY, 
+    SMXAI_CHAT_INSTRUCTIONS, 
+    SMXAI_WEBSITE_DESCRIPTION,
+)
+from syntaxmatrix.settings.client_items import read_client_file, getenv_api_key
 
 # ──────── framework‐local storage paths ────────
 # this ensures the key & data always live under the package dir,
-# regardless of where the developer `cd` into before launching.
 _CLIENT_DIR = detect_project_root()
 _HISTORY_DIR   = os.path.join(_CLIENT_DIR, "smx_history")
 os.makedirs(_HISTORY_DIR, exist_ok=True)
 
+_BRANDING_DIR = os.path.join(_CLIENT_DIR, "branding")
+os.makedirs(_BRANDING_DIR, exist_ok=True)
+
 _SECRET_PATH = os.path.join(_CLIENT_DIR, ".smx_secret_key")
 
-_CLIENT_DOTENV_PATH = os.path.join(str(_CLIENT_DIR.parent), ".env")
-if os.path.isfile(_CLIENT_DOTENV_PATH):
-    load_dotenv(_CLIENT_DOTENV_PATH, override=True)
+# OPENAI_API_KEY = getenv_api_key(_CLIENT_DIR, "OPENAI_API_KEY"))
+# dotenv_content = read_client_file(_CLIENT_DIR, ".env")
 
 _ICONS_PATH = os.path.join(_CLIENT_DIR, "static", "icons")
 os.makedirs(_ICONS_PATH, exist_ok=True)
@@ -53,9 +58,9 @@ class SyntaxMUI:
            host="127.0.0.1", 
             port="5080", 
             user_icon="👩🏿‍🦲",
-            bot_icon="<img src='/static/icons/favicon.png' width=20' alt='bot'/>",
+            bot_icon="<img src='/static/icons/bot-icon.png' width=20' alt='bot'/>",
             favicon="/static/icons/favicon.png",      
-            site_logo="<img src='/static/icons/logo.png' width='30' alt='logo'/>",           
+            site_logo="<img src='/static/icons/logo.png' width='45' alt='logo'/>",           
             site_title="SyntaxMatrix", 
             project_name="smxAI", 
             theme_name="light",
@@ -70,12 +75,15 @@ class SyntaxMUI:
         self.bot_icon = bot_icon
         self.site_logo = site_logo
         self.favicon = favicon
+        self._default_site_logo = site_logo
+        self._default_favicon = favicon
         self.site_title = site_title
         self.project_name = project_name
         self.ui_mode = ui_mode
         self.theme_toggle_enabled = False
         self.user_files_enabled = False
-        self.smxai_identity = SMXAI_CHAT_ID
+        self.registration_enabled = False
+        self.smxai_identity = SMXAI_CHAT_IDENTITY
         self.smxai_instructions = SMXAI_CHAT_INSTRUCTIONS    
         self.website_description = SMXAI_WEBSITE_DESCRIPTION
         self._eda_output = {}      # {chat_id: html}
@@ -96,15 +104,23 @@ class SyntaxMUI:
         self._last_llm_usage = None 
         routes.setup_routes(self)
 
-        self._admin_profile = {}
-        self._chat_profile = {}
-        self._coding_profile = {}
-        self._classification_profile = {}
-        self._summarization_profile = {}
+        # Apply client branding overrides if present on disk
+        self._apply_branding_from_disk()
+
+        # LLM Profiles
+        self.admin_profile = {}
+        self.chat_profile = {}
+        self.classifier_profile = {}
+        self.summarizer_profile = {}
+        self.coder_profile = {}
+        self.imagetexter_profile = {}
+        self.textimager_profile = {}
+        self.imageeditor_profile = {}
             
         self._gpt_models_latest_prev_resp_ids = {}
         self.is_streaming = False
         self.stream_args = {}
+        self._apply_feature_flags_from_db()
 
         self._recent_visual_summaries = []
 
@@ -311,6 +327,27 @@ class SyntaxMUI:
     
     def enable_user_files(self):
         self.user_files_enabled = True
+
+    def enable_registration(self):
+        self.registration_enabled = True
+
+    def _apply_feature_flags_from_db(self):
+        """
+        Pull persisted toggles from app_settings.
+        """
+        def _truthy(v):
+            return str(v or "").strip().lower() in ("1", "true", "yes", "on")
+
+        try:
+            stream_v = db.get_setting("feature.stream_mode", "0")
+            user_files_v = db.get_setting("feature.user_files", "0")
+
+            self.is_streaming = _truthy(stream_v)
+            self.user_files_enabled = _truthy(user_files_v)
+        except Exception:
+            # Keep defaults if DB isn't ready for any reason
+            pass
+
     
     @staticmethod
     def columns(components):
@@ -320,23 +357,49 @@ class SyntaxMUI:
         col_html += "</div>"
         return col_html
 
+    # Site Branding
     def set_site_title(self, title):
         self.site_title = title
-    
     def set_project_name(self, project_name):
         self.project_name = project_name
-
     def set_favicon(self, icon):
         self.favicon = icon
-
     def set_site_logo(self, logo):
         self.site_logo = logo
-
     def set_user_icon(self, icon):
         self.user_icon = icon
-
     def set_bot_icon(self, icon):
         self.bot_icon = icon
+    
+    def _apply_branding_from_disk(self):
+        """
+        If a client logo/favicon exists in syntaxmatrixdir/branding/,
+        use it; otherwise keep the framework defaults.
+        """
+        branding_dir = os.path.join(_CLIENT_DIR, "branding")
+
+        def _pick(basename: str):
+            for ext in (".png", ".jpg", ".jpeg"):
+                fn = f"{basename}{ext}"
+                p = os.path.join(branding_dir, fn)
+                if os.path.exists(p):
+                    return fn
+            return None
+
+        logo_fn = _pick("logo")
+        fav_fn = _pick("favicon")
+
+        if logo_fn:
+            # Use client-served endpoint (added in routes.py below)
+            self.site_logo = f"<img src='/branding/{logo_fn}' width='45' alt='logo'/>"
+        else:
+            self.site_logo = getattr(self, "_default_site_logo", self.site_logo)
+
+        if fav_fn:
+            self.favicon = f"/branding/{fav_fn}"
+        else:
+            self.favicon = getattr(self, "_default_favicon", self.favicon)
+
 
     def text_input(self, key, id, label, placeholder=""):
         if not placeholder:
@@ -514,12 +577,12 @@ class SyntaxMUI:
     # ──────────────────────────────────────────────────────────────
     #  *********** LLM CLIENT HELPERS  **********************
     # ──────────────────────────────────────────────────────────────
-    def set_prompt_profile(self, profile):
-        self.ai_chat_id = profile
+    def set_smxai_identity(self, profile):
+        self.set_smxai_identity = profile
     
 
-    def set_prompt_instructions(self, instructions):
-        self.ai_chat_instructions = instructions
+    def set_smxai_instructions(self, instructions):
+        self.set_smxai_instructions = instructions
 
 
     def set_website_description(self, desc):
@@ -592,18 +655,17 @@ class SyntaxMUI:
 
 
     def classify_query_intent(self, query: str) -> str:
-        from syntaxmatrix.gpt_models_latest import extract_output_text as _out, set_args 
    
-        if not self._classification_profile:
-            classification_profile = _prof.get_profile('classification') or _prof.get_profile('chat') or _prof.get_profile('admin')
-            if not classification_profile:
-                return {"Error": "Set a profile for Classification"}
-            self._classification_profile = classification_profile
-            self._classification_profile['client'] = _prof.get_client(classification_profile)
+        if not self.classifier_profile:
+            classifier_profile = _prof.get_profile('classifier') or _prof.get_profile('chat') or _prof.get_profile('summarizer') or _prof.get_profile('admin')
+            if not classifier_profile:
+                return "Error: Set a profile for Classification"
+            self.classifier_profile = classifier_profile
+            self.classifier_profile['client'] = _prof.get_client(classifier_profile)
 
-        _client = self._classification_profile['client']
-        _provider = self._classification_profile['provider']
-        _model = self._classification_profile['model']
+        _client = self.classifier_profile['client']
+        _provider = self.classifier_profile['provider']
+        _model = self.classifier_profile['model']
 
         # New instruction format with hybrid option
         _intent_profile = "You are an intent classifier. Respond ONLY with the intent name."
@@ -697,35 +759,36 @@ class SyntaxMUI:
 
     def generate_contextual_title(self, chat_history):
         
-        if not self._summarization_profile:
-            summarization_profile = _prof.get_profile('summarization') or _prof.get_profile('chat') or _prof.get_profile('admin') 
-            if not summarization_profile:
-                return {"Error": "Chat profile not set yet."}
+        if not self.summarizer_profile:
+            summarizer_profile = _prof.get_profile('summarizer') or _prof.get_profile('classifier') or _prof.get_profile('chat') or _prof.get_profile('admin') 
+            if not summarizer_profile:
+                return "<code style='color:red;'>Error: No Agent setup yet.</code>"
             
-            self._summarization_profile = summarization_profile
-            self._summarization_profile['client'] = _prof.get_client(summarization_profile)
+            self.summarizer_profile = summarizer_profile
+            self.summarizer_profile['client'] = _prof.get_client(summarizer_profile)
 
         conversation = "\n".join([f"{role}: {msg}" for role, msg in chat_history])
         _title_profile = "You are a title generator that creates concise and relevant titles for the given conversations."
+
         _instructions = f"""
             Generate a contextual title (5 short words max) from the given Conversation History 
             The title should be concise - with no preamble, relevant, and capture the essence of this Conversation: \n{conversation}.\n\n
             return only the title.
         """
         
-        _client = self._summarization_profile['client']
-        _provider = self._summarization_profile['provider']
-        _model = self._summarization_profile['model']
+        _client = self.summarizer_profile['client']
+        _provider = self.summarizer_profile['provider']
+        _model = self.summarizer_profile['model']
 
         def google_generated_title():
             try:
                 response = _client.models.generate_content(
                     model=_model,
-                    contents=f"{_title_profile}\n{_instructions}"
+                    contents=f"{_title_profile}\n\n{_instructions}"
                 )
                 return response.text.strip()
             except Exception as e:
-                return f"Summary agent error!"
+                return f"Google Summary agent error!"
 
         def gpt_models_latest_generated_title():
             try:                 
@@ -740,7 +803,7 @@ class SyntaxMUI:
                 resp = _client.responses.create(**args)
                 return _out(resp).strip()
             except Exception as e:
-                return f"Summary agent error!"
+                return f"OpenAI 5s Summary agent error!"
         
         def anthropic_generated_title():       
             try:
@@ -753,7 +816,7 @@ class SyntaxMUI:
                 )
                 return response.content[0].text.strip()  
             except Exception as e:
-                return f"Summary agent error!"
+                return f"Anthropic Summary agent error!"
             
         def openai_sdk_generated_title():     
             prompt = [
@@ -767,10 +830,14 @@ class SyntaxMUI:
                     temperature=0.3,
                     max_tokens=50
                 ) 
-                title = response.choices[0].message.content.strip().lower()
-                return title if title else ""
+                print("\nRESPONSE:\n", response)
+
+                title = response.choices[0].message.content
+
+                print("\nTITLE:\n", title)
+                return title 
             except Exception as e:
-               return f"Summary agent error!"
+               return f"SDK Summary agent error!"
 
         if _provider == "google":
             title = google_generated_title()
@@ -792,18 +859,18 @@ class SyntaxMUI:
 
     def process_query_stream(self, query: str, context: str, history: list, stream=True) -> Generator[str, None, None]:
        
-        if not self._chat_profile:
+        if not self.chat_profile:
             chat_profile = _prof.get_profile("chat") or _prof.get_profile("admin")
             if not chat_profile:
                 yield """<p style='color:red;'>Error: Chat profile is not configured. Add a chat profile inside the admin panel or contact your administrator.</p>
                 """
                 return None
-            self._chat_profile = chat_profile
-            self._chat_profile['client'] = _prof.get_client(chat_profile)
+            self.chat_profile = chat_profile
+            self.chat_profile['client'] = _prof.get_client(chat_profile)
 
-        _provider = self._chat_profile['provider']
-        _client = self._chat_profile['client']
-        _model = self._chat_profile['model']
+        _provider = self.chat_profile['provider']
+        _client = self.chat_profile['client']
+        _model = self.chat_profile['model']
 
         _contents = f"""
             {self.smxai_instructions}\n\n 
@@ -877,18 +944,18 @@ class SyntaxMUI:
     
     def process_query(self, query, context, history, stream=False):
 
-        if not self._chat_profile:
+        if not self.chat_profile:
             chat_profile = _prof.get_profile("chat") or _prof.get_profile("admin")
             if not chat_profile:
                 return """<p style='color:red;'>Error: Chat profile is not configured. Add a chat profile inside the admin panel or contact your administrator.</p>
                 """
                 return
                  
-            self._chat_profile = chat_profile
-            self._chat_profile['client'] = _prof.get_client(chat_profile) 
-        _provider = self._chat_profile['provider']
-        _client = self._chat_profile['client']
-        _model = self._chat_profile['model']
+            self.chat_profile = chat_profile
+            self.chat_profile['client'] = _prof.get_client(chat_profile) 
+        _provider = self.chat_profile['provider']
+        _client = self.chat_profile['client']
+        _model = self.chat_profile['model']
         _contents = f"""
                     {self.smxai_instructions}\n\n
                     Question: {query}\n
@@ -1027,7 +1094,7 @@ class SyntaxMUI:
             return code
         _prompt = f"```python\n{code}\n```"
                   
-        repair_profile = _prof.get_profile("vision2text") or _prof.get_profile("admin")
+        repair_profile = self.coder_profile 
         if not repair_profile:
             return (
                 '<div class="smx-alert smx-alert-warn">'
@@ -1137,21 +1204,28 @@ class SyntaxMUI:
         tasks = [str(t).strip().lower() for t in tasks if str(t).strip()]
 
         ai_profile = """
-        - You are a Python expert specializing in data science and machine learning. 
-        - Your task is to generate a single, complete, production-quality, executable Python script for a Jupyter-like Python kernel, based on the given instructions. 
+        - You are a Python expert specialising in Data Science (DS) and Machine Learning (ML).
+        - Your task is to generate a single, complete, production-ready Python script that can be executed in a Jupyter-like Python kernel, based on the given instructions.
         - The dataset is already loaded as a pandas DataFrame named `df` (no file I/O or file uploads).
-        - Make a copy of `df` and name it `df_copy`. Make sure `df_copy` is preprocessed and cleaned, named `df_cleaned`, if not already done so. Then use `df_cleaned` to perform the ML tasks described in the given context.
-        - Select your features and target, from `df_cleaned`, with care and name it `required_cols`
-        - Create your 'df_filtered by doing: df_filtered = df_cleaned[required_cols].
-        - Use the {TEMPLATE_CATALOGUE} below to educate yourself on which visualizations you will implement in the code.
-        - The final output MUST be the complete, executable Python code only, enclosed in a single markdown code block (```python ... ```), which is required to fulfill the user's request. See the {tasks} below.
+        - Make a copy of `df` and name it `df_copy`.
+        - Make sure `df_copy` is preprocessed and cleaned, and name it `df_cleaned`, if not already done so.
+        - Work only with `df_cleaned` to perform the ML tasks described in the given context.
+
+        - Always treat modelling as features `X` and target `y`:
+            * Choose ONE target column in `df_cleaned` (the value to be predicted) and refer to it as `target_col` or `y`.
+            * Build the feature matrix `X` from `df_cleaned` WITHOUT including the target column or any direct transformation of it.
+            * Examples of forbidden feature leakage: if predicting `sellingprice`, do NOT include `sellingprice`, `log_sellingprice`, `margin = sellingprice - mmr`, or any other direct function of `sellingprice` in `X`.
+            * You may create target-derived columns (margins, flags, percentage differences) for descriptive tables or plots, but NEVER use them as model inputs.
+
+        - When you need a modelling frame, define `required_cols = [target_col] + feature_cols` where `feature_cols` excludes the target and its transforms, and then create `df_filtered = df_cleaned[required_cols]`.
+
+        - Use the {TEMPLATE_CATALOGUE} below to educate yourself about available helper functions and reference code, and ensure the implementations are in the code you generate.
+        - The final output MUST BE the complete, executable Python code for the requested analysis, wrapped in a single fenced Python code block (```python ... ```), and MUST BE able to fulfil the user's request: {tasks}.
         - Do not include any explanatory text or markdown outside the code block.
         """
 
         TEMPLATE_CATALOGUE = """
-        ### Available SyntaxMatrix templates (use these instead of inventing new helpers)
-
-        Visualisation templates (dataset-agnostic):
+        Visualisation templates:
         - viz_pie(df, category_col=None, top_k=8): pie/donut shares within a category.
         - viz_stacked_bar(df, x=None, hue=None, normalise=True): composition across groups.
         - viz_count_bar(df, category_col=None, top_k=12): counts/denominators by category.
@@ -1191,9 +1265,9 @@ class SyntaxMUI:
             
             """
             ### Template rules
-            - You MAY call a template if it matches the task.
+            - You MAY call 1 or more templates if they matche the task.
             - Do NOT invent template names.
-            - If no template fits, write minimal direct pandas/sklearn/seaborn code instead.
+            - If no template fits, write minimal direct pandas/sklearn/seaborn code instead, for visualization.
             - Keep the solution short: avoid writing wrappers/utilities already handled by SyntaxMatrix hardener.
 
             #### Template selection hint examples:
@@ -1216,8 +1290,7 @@ class SyntaxMUI:
                 set `random_state=42` where relevant.
             4) Be defensive, but avoid hard-failing on optional fields:
             - If the primary column, needed to answer the question, is missing, review your copy of the `df` again.
-            Make sure that you selected the proper column. 
-            Never use a column/variable which isn't available or defined.
+            - Make sure that you selected the proper column. Never use a column/variable which isn't available or defined.
             - If a secondary/extra column is missing, show a warning with `show(...)` and continue using available fields.
             - Handle missing values sensibly (drop rows for simple EDA; use `ColumnTransformer` + `SimpleImputer` for modelling).
             - For categorical features in ML, use `OneHotEncoder(handle_unknown="ignore")`
@@ -1249,6 +1322,20 @@ class SyntaxMUI:
             11) You MUST NOT reference any column outside Available columns: {AVAILABLE_COLUMNS}.
             12) If asked to predict/classify, choose the target by matching the task text to Allowed columns
                 and never invent a new name.
+            13) Treat df as the primary dataset you must work with.
+            14) The dataset is already loaded as df (no file I/O or file uploads).
+            15) All outputs must be visible to the user via the provided show(...) helper.
+            16) Never use print(...); use show(...) instead.
+            17) You MUST NOT read from or write to local files, folders, or external storage.
+            - Do not call open(...), Path(...).write_text/write_bytes, or similar file APIs.
+            - Do not use df.to_csv(...), df.to_excel(...), df.to_parquet(...),
+                df.to_pickle(...), df.to_json(...), df.to_hdf(...), or any other
+                method that writes to disk.
+            - Do not call joblib.dump(...), pickle.dump(...), torch.save(...),
+                numpy.save(...), numpy.savetxt(...), or similar saving functions.
+            - Do not call plt.savefig(..., 'somefile.png') or any variant that
+                writes an image to a filename. Plots must be rendered in-memory only.
+            18) Keep everything in memory and surface results via show(...) or plots.
             
             #### Cohort rules
             When you generate plots for cohorts or categories, you MUST obey these rules:
@@ -1297,21 +1384,21 @@ class SyntaxMUI:
             - And ends with at least 3 visible output (`show(...)` and/or `plt.show()`).
         """)
 
-        if not self._coding_profile:
-            coding_profile = _prof.get_profile("coding") or _prof.get_profile("admin")
-            if not coding_profile:
+        if not self.coder_profile:
+            _coder_profile = _prof.get_profile("coder")  
+            if not _coder_profile:
                 return (
                     '<div class="smx-alert smx-alert-warn">'
-                        'No LLM profile configured for <code>coding</code> (or <code>admin</code>). <br>'
+                        'No LLM profile configured for <code>coding</code> <br>'
                         'Please, add the LLM profile inside the admin panel or contact your Administrator.'
                     '</div>'
                 )
 
-            self._coding_profile = coding_profile
-            self._coding_profile['client'] = _prof.get_client(coding_profile)
+            self.coder_profile = _coder_profile
+            self.coder_profile['client'] = _prof.get_client(_coder_profile)
 
         # code = mlearning_agent(instructions, ai_profile, self._coding_profile)
-        code, usage = mlearning_agent(instructions, ai_profile, self._coding_profile)
+        code, usage = mlearning_agent(instructions, ai_profile, self.coder_profile)
         self._last_llm_usage = usage
 
         if code:
@@ -1359,7 +1446,22 @@ class SyntaxMUI:
             return code.strip()
 
         return "Error: AI code generation failed."
+    
 
+    def get_image_generator_profile(self):
+        if not self._fullvision_profile:
+            fullvision_profile = _prof.get_profile("fullvision")
+            if not fullvision_profile:
+                return (
+                    '<div class="smx-alert smx-alert-warn">'
+                        'No Full Vision profile configured for <code>coding</code> <br>'
+                        'Please, add it inside the admin panel or contact your Administrator.'
+                    '</div>'
+                )          
+            self._fullvision_profile = fullvision_profile
+            self._fullvision_profile['client'] = _prof.get_client(fullvision_profile) 
+
+        return self._fullvision_profile
 
     def sanitize_rough_to_markdown_task(self, rough: str) -> str:
         """
@@ -1395,7 +1497,13 @@ class SyntaxMUI:
 
         # Drop optional <rough> wrapper
         return out.replace("<rough>", "").replace("</rough>", "").strip()
-
+        
+    def current_profile(self, agency):
+        current_profile = _prof.get_profile(agency) or _prof.get_profile('admin') 
+        if not current_profile:
+            return "Error: Configure the valid LLM profile."
+        current_profile['client'] = _prof.get_client(current_profile)
+        return current_profile
 
     def run(self):
         url = f"http://{self.host}:{self.port}/"