update

quote_generator_gemini.py

@@ -27,30 +27,42 @@ class QuoteGenerator:
             raise ValueError("GEMINI_API_KEY not found in environment variables")
         
         genai.configure(api_key=api_key)
-        self.model = genai.GenerativeModel('gemini-3-pro-preview')  # Updated model name
+        self.model = genai.GenerativeModel('gemini-1.5-flash')  # Updated model name
         
-        # Persistent storage for quote history
+        # Persistent storage for quote history - PER NICHE
         self.history_file = history_file
-        self.recent_quotes = self._load_history()
-        self.max_history = 100  # Keep last 100 quotes
+        self.quotes_by_niche = self._load_history()  # Returns dict
+        self.max_history = 100  # Keep last 100 quotes PER NICHE
         
-    def _load_history(self) -> List[str]:
-        """Load quote history from persistent storage"""
+    def _load_history(self) -> dict:
+        """Load quote history from persistent storage - organized by niche"""
         try:
             if os.path.exists(self.history_file):
                 with open(self.history_file, 'r') as f:
                     data = json.load(f)
-                    return data.get('quotes', [])
+                    # Support both old format (list) and new format (dict)
+                    if 'quotes_by_niche' in data:
+                        print(f"📖 Loaded history for {len(data['quotes_by_niche'])} niches")
+                        return data['quotes_by_niche']
+                    elif 'quotes' in data:
+                        # Old format - migrate
+                        print("📖 Migrating to per-niche tracking")
+                        return {"General": data['quotes']}
         except Exception as e:
             print(f"Could not load history: {e}")
-        return []
+        return {}
     
     def _save_history(self):
         """Save quote history to persistent storage"""
         try:
             os.makedirs(os.path.dirname(self.history_file), exist_ok=True)
+            # Save all niches, keeping last max_history per niche
+            trimmed_data = {}
+            for niche, quotes in self.quotes_by_niche.items():
+                trimmed_data[niche] = quotes[-self.max_history:]
+            
             with open(self.history_file, 'w') as f:
-                json.dump({'quotes': self.recent_quotes[-self.max_history:]}, f)
+                json.dump({'quotes_by_niche': trimmed_data}, f, indent=2)
         except Exception as e:
             print(f"Could not save history: {e}")
     
@@ -66,34 +78,47 @@ class QuoteGenerator:
             A unique quote string
         """
         
-        # Get recent quotes to avoid (last 20)
+        # Get this niche's history
+        if niche not in self.quotes_by_niche:
+            self.quotes_by_niche[niche] = []
+        
+        recent_quotes = self.quotes_by_niche[niche]
+        
+        # Get recent quotes to avoid (last 30 for better variety)
         recent_quotes_text = ""
-        if self.recent_quotes:
-            recent_quotes_text = "\n\nPREVIOUSLY GENERATED (DO NOT REPEAT OR PARAPHRASE THESE):\n"
-            for i, quote in enumerate(self.recent_quotes[-20:], 1):
+        if recent_quotes:
+            recent_quotes_text = f"\n\n❌ PREVIOUSLY GENERATED {niche.upper()} QUOTES (YOU MUST AVOID THESE - DO NOT REPEAT OR PARAPHRASE):\n"
+            for i, quote in enumerate(recent_quotes[-30:], 1):  # Last 30
                 recent_quotes_text += f"{i}. {quote}\n"
+            
+            print(f"📊 {niche} history: {len(recent_quotes)} total quotes, showing last 30 to avoid")
+        else:
+            print(f"📊 No {niche} history yet - generating first quote for this niche")
         
-        # Build prompt with variety instructions
-        prompt = f"""Generate a UNIQUE and powerful {niche} quote suitable for an Instagram/TikTok video.
+        # Build prompt with VERY strong variety instructions
+        prompt = f"""Generate a COMPLETELY UNIQUE and powerful {niche} quote suitable for an Instagram/TikTok video.
 
 Visual Style: {style}
 
-CRITICAL REQUIREMENTS:
+CRITICAL REQUIREMENTS - READ CAREFULLY:
 - 2-4 sentences maximum (can be longer if deeply meaningful)
-- Must be COMPLETELY DIFFERENT from all previously generated quotes
+- Must be RADICALLY DIFFERENT from all previously generated quotes below
 - Inspirational and impactful
-- Deep and meaningful insights
+- Deep and meaningful insights  
 - Fresh perspective that viewers haven't heard before
 - Unexpected wisdom or unique angle
 - DO NOT use clichés or overused phrases
+- AVOID common motivational phrases like "success is", "believe in yourself", "the only limit"
 
 {recent_quotes_text}
 
-IMPORTANT: Your quote must be COMPLETELY ORIGINAL and DIFFERENT in:
-- Core message and theme
-- Wording and phrasing
-- Perspective and angle
-- Examples or metaphors used
+IMPORTANT: Your quote must be COMPLETELY ORIGINAL in:
+- Core message and theme (different life area or concept)
+- Wording and phrasing (no similar sentences)
+- Perspective and angle (unique viewpoint)
+- Examples or metaphors used (creative analogies)
+
+Generate something viewers have NEVER seen before.
 
 Return ONLY the quote text, nothing else. No quotation marks, no attribution."""
 
@@ -115,26 +140,53 @@ Return ONLY the quote text, nothing else. No quotation marks, no attribution."""
             quote = response.text.strip()
             quote = quote.strip('"').strip("'").strip()
             
-            # Add to history and save
-            self.recent_quotes.append(quote)
+            # Check if this quote already exists in THIS niche
+            if quote in recent_quotes:
+                print(f"⚠️ WARNING: Generated duplicate {niche} quote! Trying again with higher temperature...")
+                # Try one more time with even higher temperature
+                response = self.model.generate_content(
+                    prompt,
+                    generation_config={
+                        "temperature": 1.2,  # Even higher
+                        "top_p": 0.98,
+                        "top_k": 80,
+                        "max_output_tokens": 200,
+                    }
+                )
+                quote = response.text.strip().strip('"').strip("'").strip()
+            
+            # Add to this niche's history and save
+            self.quotes_by_niche[niche].append(quote)
             self._save_history()
             
+            print(f"✅ Generated unique {niche} quote #{len(self.quotes_by_niche[niche])}")
+            print(f"💾 Saved to: {self.history_file}")
+            
             return quote
         
         except Exception as e:
             raise Exception(f"Gemini quote generation failed: {str(e)}")
     
     def get_stats(self) -> dict:
-        """Get statistics about quote generation"""
+        """Get statistics about quote generation - per niche"""
+        total_quotes = sum(len(quotes) for quotes in self.quotes_by_niche.values())
         return {
-            "total_quotes_generated": len(self.recent_quotes),
-            "unique_quotes": len(set(self.recent_quotes)),
-            "history_size": len(self.recent_quotes[-self.max_history:])
+            "total_quotes_generated": total_quotes,
+            "quotes_by_niche": {niche: len(quotes) for niche, quotes in self.quotes_by_niche.items()},
+            "niches_tracked": len(self.quotes_by_niche)
         }
     
-    def clear_history(self):
-        """Clear quote history (use with caution)"""
-        self.recent_quotes = []
+    def clear_history(self, niche: Optional[str] = None):
+        """Clear quote history (use with caution)
+        
+        Args:
+            niche: If provided, clear only this niche. Otherwise clear all.
+        """
+        if niche:
+            if niche in self.quotes_by_niche:
+                self.quotes_by_niche[niche] = []
+        else:
+            self.quotes_by_niche = {}
         self._save_history()
 
 
@@ -278,10 +330,9 @@ if __name__ == "__main__":
         stats = generator.get_stats()
         print("\nStats:")
         print(f"  Total generated: {stats['total_quotes_generated']}")
-        print(f"  Unique quotes: {stats['unique_quotes']}")
-        print(f"  History size: {stats['history_size']}")
+        print(f"  Niches tracked: {stats['niches_tracked']}")
+        print(f"  Per niche: {stats['quotes_by_niche']}")
         
     except Exception as e:
         print(f"Error: {e}")
         print("\nMake sure GEMINI_API_KEY is set in environment variables")
-