Tech Debt: Migrate to unified google-genai SDK

agents/memory_agent.py

@@ -2,7 +2,7 @@ import os
 import time
 from dotenv import load_dotenv
 from pinecone import Pinecone, ServerlessSpec
-import google.generativeai as genai
+from google import genai
 
 load_dotenv()
 
@@ -12,7 +12,7 @@ class MemoryAgent:
         self.gemini_api_key = os.getenv("GEMINI_API_KEY")
         if not self.gemini_api_key:
             raise ValueError("GEMINI_API_KEY not found")
-        genai.configure(api_key=self.gemini_api_key)
+        self.client = genai.Client(api_key=self.gemini_api_key)
         
         # Configure Pinecone
         self.pinecone_api_key = os.getenv("PINECONE_API_KEY")
@@ -44,12 +44,12 @@ class MemoryAgent:
 
     def _get_embedding(self, text):
         # Using models/text-embedding-004
-        result = genai.embed_content(
-            model="models/text-embedding-004",
-            content=text,
-            task_type="retrieval_document"
+        result = self.client.models.embed_content(
+            model="text-embedding-004",
+            contents=text,
+            config={"task_type": "retrieval_document"}
         )
-        return result['embedding']
+        return result.embeddings[0].values
 
     def seed_database(self):
         # Check if empty

agents/visual_analyst.py

@@ -1,7 +1,7 @@
 import os
 import json
 import re
-import google.generativeai as genai
+from google import genai
 from dotenv import load_dotenv
 
 load_dotenv()
@@ -12,9 +12,8 @@ class VisualAnalyst:
         if not self.api_key:
             raise ValueError("GEMINI_API_KEY not found")
         
-        genai.configure(api_key=self.api_key)
-        self.model_name = "models/gemini-flash-latest"
-        self.model = genai.GenerativeModel(self.model_name)
+        self.client = genai.Client(api_key=self.api_key)
+        self.model_name = "gemini-1.5-flash"
         print(f"✅ VisualAnalyst stored Gemini model: {self.model_name}")
 
     async def analyze_image(self, image_path: str):
@@ -35,9 +34,12 @@ class VisualAnalyst:
                 "Return ONLY valid JSON with keys: main_color, product_type, design_style, visual_features."
             )
             
-            # Gemini 1.5 Flash supports JSON response schema, but simple prompting often works well too.
             # We'll stick to prompt engineering for now to match the "Return ONLY valid JSON" instruction.
-            response = self.model.generate_content([user_prompt, img], request_options={'timeout': 15.0})
+            response = self.client.models.generate_content(
+                model=self.model_name,
+                contents=[user_prompt, img],
+                config={'timeout': 15.0} # Alternatively, timeouts might be configured at the client level, but we maintain the logic.
+            )
             
             response_text = response.text
             

fix_browse_button.py

@@ -0,0 +1,57 @@
+import sys
+import subprocess
+import re
+
+def fix_html():
+    with open('dashboard.html', 'r', encoding='utf-8') as f:
+        content = f.read()
+
+    js_snippet = """
+        const browseBtn = document.getElementById('browseBtn');
+        browseBtn.addEventListener('click', (e) => {
+            e.preventDefault();
+            fileInput.click();
+        });
+        fileInput.addEventListener('change', () => {
+            if (fileInput.files.length > 0) {
+                // Provide a visual cue that a file was selected
+                const fileName = fileInput.files[0].name;
+                browseBtn.innerHTML = `<span class="material-symbols-outlined text-xl">check_circle</span> ${fileName}`;
+            }
+        });
+"""
+
+    if "browseBtn.addEventListener('click', (e) => {" in content and "Provide a visual cue that a file was selected" in content:
+        print("Snippet already exists. Skipping injection.")
+    else:
+        # Find where the DOM elements are defined
+        target = "const downloadBtn = document.getElementById('downloadBtn');"
+        if target in content:
+            new_content = content.replace(target, target + "\n" + js_snippet)
+            with open('dashboard.html', 'w', encoding='utf-8') as f:
+                f.write(new_content)
+            print("Injected JS successfully.")
+        else:
+            print("Could not find insertion point!")
+            return False
+            
+    # Run git commands
+    subprocess.run(['git', 'add', 'dashboard.html'], check=True)
+    try:
+        subprocess.run(['git', 'commit', '-m', 'Bugfix: Wire up Browse Files button to hidden input'], check=True)
+    except subprocess.CalledProcessError:
+        print("Nothing to commit")
+        
+    subprocess.run(['git', 'push', '--force', 'space', 'HEAD:main'], check=True)
+    
+    # Push to origin main as well
+    try:
+        subprocess.run(['git', 'push', 'origin', 'HEAD:main'], check=True)
+    except subprocess.CalledProcessError:
+        print("Push to origin failed or not needed")
+        
+    print("Deployment triggered successfully.")
+    return True
+
+if __name__ == '__main__':
+    fix_html()

fix_dashboard_api.py

@@ -0,0 +1,66 @@
+import sys
+import subprocess
+import re
+
+def patch_dashboard():
+    with open('dashboard.html', 'r', encoding='utf-8') as f:
+        content = f.read()
+    
+    # Locate the setTimeout block inside startBtn.addEventListener and replace it with fetch logic
+    pattern = re.compile(r"setTimeout\(\(\) => \{[\s\S]*?\}, 1500\);", re.DOTALL)
+    
+    new_block = """try {
+                const formData = new FormData();
+                formData.append('file', fileInput.files[0]);
+                
+                const response = await fetch('/generate-catalog', {
+                    method: 'POST',
+                    body: formData
+                });
+                
+                const data = await response.json();
+                jsonOutput.textContent = JSON.stringify(data, null, 2);
+                isCatalogGenerated = true;
+            } catch (error) {
+                console.error("Error generating catalog:", error);
+            } finally {
+                startBtn.innerHTML = '<div class="absolute inset-0 flex items-center justify-center gap-2 lg:gap-3 relative z-10"><span class="text-white text-base lg:text-lg font-bold tracking-wide group-hover:scale-105 transition-transform">Start Agent Workflow</span><span class="material-symbols-outlined text-white text-lg lg:text-xl group-hover:translate-x-1 transition-transform">arrow_forward</span></div>';
+                startBtn.disabled = false;
+                startBtn.classList.add('animate-pulse-slow', 'animate-glow-pulse');
+            }"""
+
+    if not pattern.search(content):
+        print("Error: Could not find the target setTimeout block in dashboard.html.")
+        return False
+
+    new_content = pattern.sub(new_block, content)
+
+    with open('dashboard.html', 'w', encoding='utf-8') as f:
+        f.write(new_content)
+    
+    print("Successfully patched dashboard.html")
+    return True
+
+def run_git_commands():
+    commands = [
+        ['git', 'add', 'dashboard.html'],
+        ['git', 'commit', '-m', 'Bugfix: Restore real API connection to Glassmorphism UI'],
+        ['git', 'push', '--force', 'space', 'HEAD:main']
+    ]
+    
+    for cmd in commands:
+        print(f"Running: {' '.join(cmd)}")
+        result = subprocess.run(cmd, capture_output=True, text=True)
+        if result.returncode != 0:
+            print(f"Command failed with {result.returncode}: \\n{result.stderr}")
+            # Don't break here, let it try the other commands just in case, though push might fail if commit fails.
+            if cmd[1] == 'commit' and "nothing to commit" in result.stdout + result.stderr:
+                continue
+            if cmd[1] == 'push':
+                pass
+        else:
+            print(f"Success!\\n{result.stdout}")
+
+if __name__ == '__main__':
+    if patch_dashboard():
+        run_git_commands()

fix_js_syntax.py

@@ -0,0 +1,147 @@
+import sys
+import subprocess
+
+def fix_html():
+    with open('dashboard.html', 'r', encoding='utf-8') as f:
+        content = f.read()
+
+    parts = content.split('</footer>')
+    if len(parts) < 2:
+        print("Could not find </footer> in dashboard.html")
+        return False
+        
+    top_part = parts[0] + '</footer>\n'
+    
+    new_script = """<script>
+    tailwind.config.theme.extend.animation = { shine: 'shine 1.5s infinite' }
+    tailwind.config.theme.extend.keyframes = {
+        shine: { '0%': { left: '-100%' }, '100%': { left: '200%' } }
+    }
+const dropZone = document.getElementById('dropZone');
+const fileInput = document.getElementById('fileInput');
+const startBtn = document.getElementById('startBtn');
+const jsonOutput = document.getElementById('jsonOutput');
+const deployBtn = document.getElementById('deployBtn');
+const copyBtn = document.getElementById('copyBtn');
+const copyIcon = document.getElementById('copyIcon');
+let selectedFile = null;
+let isCatalogGenerated = false;
+const defaultDropZoneContent = `
+    <div class="absolute w-16 h-16 lg:w-20 lg:h-20 bg-amber-500/10 rounded-full blur-xl group-hover:bg-amber-500/20 transition-all"></div>
+    <div class="size-14 lg:size-16 relative z-10 rounded-2xl bg-gradient-to-br from-neutral-800 to-black border border-white/10 shadow-lg flex items-center justify-center transition-transform group-hover:scale-110 duration-300">
+        <span class="material-symbols-outlined text-2xl lg:text-3xl text-amber-500">cloud_upload</span>
+    </div>
+    <div class="flex flex-col items-center gap-1 relative z-10">
+        <p class="text-white text-base lg:text-lg font-bold leading-tight tracking-tight text-center">Drop Product Image Here</p>
+        <p class="text-neutral-500 text-xs lg:text-sm font-medium text-center">Supports JPG, PNG, WEBP</p>
+    </div>
+    <button id="browseBtn" class="mt-2 relative z-10 flex items-center justify-center rounded-full h-8 lg:h-9 px-4 lg:px-5 bg-white/5 hover:bg-white/10 border border-white/10 text-white text-[10px] lg:text-xs font-bold transition-all uppercase tracking-wide">
+        Browse Files
+    </button>
+`;
+function initDropZone() {
+    const currentBrowseBtn = document.getElementById('browseBtn');
+    if (currentBrowseBtn) {
+        currentBrowseBtn.addEventListener('click', (e) => {
+            e.preventDefault(); e.stopPropagation(); fileInput.click();
+        });
+    }
+}
+initDropZone();
+fileInput.addEventListener('change', (e) => {
+    if (e.target.files.length > 0) handleFile(e.target.files[0]);
+});
+dropZone.addEventListener('dragover', (e) => {
+    e.preventDefault(); dropZone.classList.add('border-amber-500', 'bg-amber-500/5');
+});
+dropZone.addEventListener('dragleave', (e) => {
+    e.preventDefault(); dropZone.classList.remove('border-amber-500', 'bg-amber-500/5');
+});
+dropZone.addEventListener('drop', (e) => {
+    e.preventDefault(); dropZone.classList.remove('border-amber-500', 'bg-amber-500/5');
+    if (e.dataTransfer.files.length > 0) {
+        fileInput.files = e.dataTransfer.files;
+        handleFile(e.dataTransfer.files[0]);
+    }
+});
+function handleFile(file) {
+    selectedFile = file;
+    dropZone.innerHTML = `
+        <div class="flex flex-col items-center justify-center gap-4 z-10">
+            <div class="size-14 lg:size-16 rounded-2xl bg-gradient-to-br from-neutral-800 to-black border border-white/10 shadow-lg flex items-center justify-center">
+                 <span class="material-symbols-outlined text-2xl lg:text-3xl text-amber-500">check_circle</span>
+            </div>
+            <div class="flex flex-col items-center gap-1">
+                <p class="text-white text-base lg:text-lg font-bold text-center">${file.name}</p>
+                <p class="text-neutral-500 text-xs lg:text-sm text-center">${(file.size / 1024).toFixed(1)} KB</p>
+            </div>
+             <button id="removeFileBtn" class="mt-2 flex items-center justify-center gap-2 rounded-full h-8 lg:h-9 px-4 lg:px-5 bg-red-500/10 hover:bg-red-500/20 text-red-400 border border-red-500/20 transition-all text-[10px] lg:text-xs font-bold uppercase tracking-wide">
+                <span class="material-symbols-outlined text-sm lg:text-base">close</span>
+                <span>Remove File</span>
+            </button>
+        </div>
+    `;
+    document.getElementById('removeFileBtn').addEventListener('click', (e) => {
+        e.stopPropagation(); e.preventDefault(); resetUploadUI();
+    });
+}
+function resetUploadUI() {
+    selectedFile = null; fileInput.value = ""; dropZone.innerHTML = defaultDropZoneContent; initDropZone();
+}
+startBtn.addEventListener('click', async (e) => {
+    e.preventDefault();
+    if (!fileInput.files || fileInput.files.length === 0) {
+        alert("Please select or drop an image first."); return;
+    }
+    const originalBtnContent = startBtn.innerHTML;
+    startBtn.innerHTML = '<div class="absolute inset-0 flex items-center justify-center gap-2 lg:gap-3"><svg class="animate-spin h-4 w-4 lg:h-5 lg:w-5 text-white" xmlns="http://www.w3.org/2000/svg" fill="none" viewBox="0 0 24 24"><circle class="opacity-25" cx="12" cy="12" r="10" stroke="currentColor" stroke-width="4"></circle><path class="opacity-75" fill="currentColor" d="M4 12a8 8 0 018-8V0C5.373 0 0 5.373 0 12h4zm2 5.291A7.962 7.962 0 014 12H0c0 3.042 1.135 5.824 3 7.938l3-2.647z"></path></svg><span class="text-white text-base lg:text-lg font-bold tracking-wide">Synthesizing...</span></div>';
+    startBtn.disabled = true; startBtn.classList.remove('animate-pulse-slow', 'animate-glow-pulse');
+    
+    try {
+        const formData = new FormData(); formData.append('file', fileInput.files[0]);
+        const response = await fetch('/generate-catalog', { method: 'POST', body: formData });
+        if (!response.ok) throw new Error("Server Error " + response.status);
+        const data = await response.json();
+        jsonOutput.textContent = JSON.stringify(data, null, 2);
+        isCatalogGenerated = true;
+    } catch (error) {
+        console.error("Agent Error:", error); alert("Pipeline failed: " + error.message);
+    } finally {
+        startBtn.innerHTML = originalBtnContent; startBtn.disabled = false; startBtn.classList.add('animate-pulse-slow', 'animate-glow-pulse');
+    }
+});
+copyBtn.addEventListener('click', () => {
+    navigator.clipboard.writeText(jsonOutput.innerText).then(() => {
+        const originalIcon = copyIcon.innerText; copyIcon.innerText = 'check'; copyIcon.classList.add('text-green-400');
+        setTimeout(() => { copyIcon.innerText = originalIcon; copyIcon.classList.remove('text-green-400'); }, 2000);
+    });
+});
+</script>
+</body>
+</html>
+"""
+
+    with open('dashboard.html', 'w', encoding='utf-8') as f:
+        f.write(top_part + new_script)
+
+    print("Replaced script successfully.")
+
+    # Run git commands
+    subprocess.run(['git', 'add', 'dashboard.html'], check=True)
+    try:
+        subprocess.run(['git', 'commit', '-m', 'Critical Bugfix: Resolve corrupted JS syntax and restore core agent loop'], check=True)
+    except subprocess.CalledProcessError:
+        print("Nothing to commit")
+        
+    subprocess.run(['git', 'push', '--force', 'space', 'HEAD:main'], check=True)
+    
+    # Push to origin main as well
+    try:
+        subprocess.run(['git', 'push', 'origin', 'HEAD:main'], check=True)
+    except subprocess.CalledProcessError:
+        print("Push to origin failed or not needed")
+        
+    print("Deployment triggered successfully.")
+
+if __name__ == '__main__':
+    fix_html()

requirements.txt

@@ -8,7 +8,7 @@ langchain-groq
 pinecone>=3.0.0
 pydantic
 python-dotenv
-google-generativeai>=0.8.3
+google-genai
 groq
 Pillow
 huggingface_hub

upgrade_docker.py

@@ -0,0 +1,51 @@
+import sys
+import subprocess
+
+def main():
+    dockerfile_path = "Dockerfile"
+    
+    # Read & Replace
+    try:
+        with open(dockerfile_path, "r", encoding="utf-8") as f:
+            lines = f.readlines()
+    except FileNotFoundError:
+        print(f"Error: Could not find {dockerfile_path}")
+        sys.exit(1)
+        
+    updated = False
+    for i, line in enumerate(lines):
+        if line.startswith("FROM python:3.9"):
+            lines[i] = "FROM python:3.11-slim\n"
+            updated = True
+            break
+            
+    if not updated:
+        print("Warning: Could not find 'FROM python:3.9' in Dockerfile. The file might already be updated or use a different base image.")
+    
+    # Save
+    try:
+        with open(dockerfile_path, "w", encoding="utf-8") as f:
+            f.writelines(lines)
+        print("Successfully updated Dockerfile.")
+    except Exception as e:
+        print(f"Error writing to Dockerfile: {e}")
+        sys.exit(1)
+        
+    # Deploy
+    print("Deploying to Hugging Face...")
+    commands = [
+        ["git", "add", "Dockerfile"],
+        ["git", "commit", "-m", "Chore: Upgrade Docker container to Python 3.11-slim"],
+        ["git", "push", "space", "clean_deploy:main"]
+    ]
+    
+    for cmd in commands:
+        print(f"Running: {' '.join(cmd)}")
+        result = subprocess.run(cmd, text=True, capture_output=True)
+        print(result.stdout)
+        if result.returncode != 0:
+            print(f"Command failed (stderr): {result.stderr}")
+            # we don't exit here so we can see all errors if any
+
+if __name__ == "__main__":
+    main()