Merge pull request #45 from stenwire/dev

Dev

Author: stenwire

backend/alembic/versions/2ca6c95dd95b_added_is_active_column.py

@@ -0,0 +1,36 @@
+"""added_is_active_column
+
+Revision ID: 2ca6c95dd95b
+Revises: bcb063b582c0
+Create Date: 2026-01-04 21:00:58.319341
+
+"""
+from typing import Sequence, Union
+
+from alembic import op
+import sqlalchemy as sa
+from sqlalchemy.dialects import postgresql
+
+# revision identifiers, used by Alembic.
+revision: str = '2ca6c95dd95b'
+down_revision: Union[str, Sequence[str], None] = 'bcb063b582c0'
+branch_labels: Union[str, Sequence[str], None] = None
+depends_on: Union[str, Sequence[str], None] = None
+
+
+def upgrade() -> None:
+    """Upgrade schema."""
+    # ### commands auto generated by Alembic - please adjust! ###
+    with op.batch_alter_table('widget_settings', schema=None) as batch_op:
+        batch_op.add_column(sa.Column('is_active', sa.Boolean(), nullable=True))
+
+    # ### end Alembic commands ###
+
+
+def downgrade() -> None:
+    """Downgrade schema."""
+    # ### commands auto generated by Alembic - please adjust! ###
+    with op.batch_alter_table('widget_settings', schema=None) as batch_op:
+        batch_op.drop_column('is_active')
+
+    # ### end Alembic commands ###

backend/app/api/widget.py

@@ -43,7 +43,9 @@ class WidgetUpdate(BaseModel):
     whatsapp_number: Optional[str] = None
     max_messages_per_session: Optional[int] = None
     max_sessions_per_day: Optional[int] = None
+    max_sessions_per_day: Optional[int] = None
     whitelisted_domains: Optional[List[str]] = None
+    is_active: Optional[bool] = None
 
 
 def normalize_url(url: str) -> str:
@@ -174,6 +176,9 @@ def update_my_widget_settings(
     if settings.whitelisted_domains is not None:
         widget.whitelisted_domains = settings.whitelisted_domains
 
+    if settings.is_active is not None:
+        widget.is_active = settings.is_active
+        
     db.commit()
     db.refresh(widget)
     db.commit()
@@ -586,6 +591,7 @@ async def process_chat_message(db: Session, widget: WidgetSettings, guest: Guest
 
 
     try:
+        print(f"Widget: Calling run_conversation with user_id={widget.user_id} (Type: {type(widget.user_id)})")
         ai_response_text = await run_conversation(
             message=message_text,
             user_id=widget.user_id,

backend/app/models/widget.py

@@ -26,6 +26,9 @@ class WidgetSettings(Base):
     whatsapp_enabled = Column(Boolean, default=False)
     whatsapp_number = Column(String, nullable=True)
 
+    # Feature Flags
+    is_active = Column(Boolean, default=True)  # Master toggle to enable/disable widget
+    
     # Denormalized Business Fields (for easier widget access)
     logo_url = Column(String, nullable=True)
 

backend/app/schemas/widget.py

@@ -68,6 +68,7 @@ class WidgetConfigResponse(BaseModel):
     max_sessions_per_day: Optional[int] = 5
     whitelisted_domains: Optional[List[str]] = None
     logo_url: Optional[str] = None
+    is_active: Optional[bool] = True
 
     class Config:
         from_attributes = True

backend/app/services/agent_system/tools.py

@@ -41,6 +41,7 @@ def get_context(user_input: str, tool_context: ToolContext) -> str:
 
     # Retrieve context with user_id and api_key
     context_chunks = rag_service.query(text=user_input, user_id=user_id, api_key=api_key)
+    print(f"--- Tool: RAG Service returned {len(context_chunks)} chunks ---")
     context_text = "\n\n".join(context_chunks)
     return context_text
 

backend/app/services/rag_service.py

@@ -167,14 +167,27 @@ def query(self, text: str, user_id: str, api_key: Optional[str] = None, db: Sess
             return []
 
         try:
-             query_embedding = self._get_query_embedding(text, api_key)
-             # Query with user_id filter and query_embedding
-             # Wrap in list as vector_db expects list of embeddings
-             results = self.vector_db.query(query_embeddings=[query_embedding], where={"user_id": user_id})
-             
-             if results and results['documents']:
-                 return results['documents'][0]
-             return []
+            print(f"RAG Service: Querying for user_id={user_id}")
+            query_embedding = self._get_query_embedding(text, api_key)
+            # Query with user_id filter and query_embedding
+            # Wrap in list as vector_db expects list of embeddings
+            
+            where_clause = {"user_id": user_id}
+            print(f"RAG Service: Executing vector_db.query with where={where_clause}")
+            
+            results = self.vector_db.query(query_embeddings=[query_embedding], where=where_clause)
+            
+            print(f"RAG Service: Raw results keys: {results.keys() if results else 'None'}")
+            if results and results.get('documents'):
+                doc_count = len(results['documents'][0])
+                print(f"RAG Service: Found {doc_count} documents.")
+                # console log first few chars of first doc if exists
+                if doc_count > 0:
+                    print(f"RAG Service: First doc snippet: {results['documents'][0][0][:50]}...")
+                return results['documents'][0]
+            
+            print("RAG Service: No documents found.")
+            return []
         except Exception as e:
             print(f"Error querying RAG: {e}")
             return []

frontend/src/app/dashboard/widget-settings/page.tsx

@@ -28,6 +28,7 @@ interface WidgetSettings {
   max_messages_per_session?: number;
   max_sessions_per_day?: number;
   whitelisted_domains?: string[];
+  is_active?: boolean;
 }
 
 type Tab = 'business' | 'appearance' | 'installation';
@@ -98,7 +99,8 @@ export default function WidgetSettingsPage() {
           whatsapp_number: settings.whatsapp_number,
           max_messages_per_session: settings.max_messages_per_session,
           max_sessions_per_day: settings.max_sessions_per_day,
-          whitelisted_domains: settings.whitelisted_domains
+          whitelisted_domains: settings.whitelisted_domains,
+          is_active: settings.is_active
         }),
       });
       if (res.ok) {
@@ -279,6 +281,24 @@ export default function WidgetSettingsPage() {
                 </div>
 
                 <div className="grid grid-cols-1 md:grid-cols-2 gap-8">
+                  <div className="md:col-span-2 bg-[var(--bg-secondary)] p-6 rounded-[var(--radius-md)] border border-[var(--border-subtle)] flex items-center justify-between">
+                    <div>
+                      <h3 className="text-sm font-bold text-[var(--text-primary)]">Widget Status</h3>
+                      <p className="text-sm text-[var(--text-secondary)] mt-1">
+                        {settings?.is_active ? 'Widget is currently active and visible on your site.' : 'Widget is disabled and hidden from your site.'}
+                      </p>
+                    </div>
+                    <label className="relative inline-flex items-center cursor-pointer">
+                      <input
+                        type="checkbox"
+                        className="sr-only peer"
+                        checked={settings?.is_active ?? true}
+                        onChange={(e) => settings && setSettings({ ...settings, is_active: e.target.checked })}
+                      />
+                      <div className="w-11 h-6 bg-gray-200 peer-focus:outline-none peer-focus:ring-4 peer-focus:ring-green-300 dark:peer-focus:ring-green-800 rounded-full peer dark:bg-gray-700 peer-checked:after:translate-x-full peer-checked:after:border-white after:content-[''] after:absolute after:top-[2px] after:left-[2px] after:bg-white after:border-gray-300 after:border after:rounded-full after:h-5 after:w-5 after:transition-all dark:border-gray-600 peer-checked:bg-[var(--brand-primary)]"></div>
+                    </label>
+                  </div>
+
                   <div>
                     <label className="block text-sm font-medium text-[var(--text-secondary)] mb-2">Primary Brand Color</label>
                     <div className="flex gap-3 items-center">

frontend/src/app/widget.js/route.ts

@@ -39,7 +39,13 @@ export async function GET() {
       if (!res.ok) throw new Error("Failed to load widget config");
       return res.json();
     })
-    .then(config => initWidget(config))
+    .then(config => {
+      if (config.is_active === false) {
+          console.log("Taimako.AI Widget: Widget is currently disabled.");
+          return;
+      }
+      initWidget(config);
+    })
     .catch(err => console.error("Taimako.AI Widget Error:", err));
 
   function initWidget(config) {