more fixes, disabled text generator and using templates now

app.py

@@ -71,55 +71,72 @@ def generate_content(
 ):
     char_limit = 280 if platform == "Twitter" else 500
     
-    # Simpler, more direct prompt
-    prompt = f"""Marketing post for {platform}:
-
-{product_name} helps {target_audience} by {unique_benefits}. {product_description}. Features include {key_features}.
-
-Marketing post in {tone} tone:
-"""
+    # Split features and benefits into lists
+    features = [f.strip() for f in key_features.split(',')]
+    benefits = [b.strip() for b in unique_benefits.split(',')]
     
-    try:
-        # Generate content with stricter parameters
-        inputs = generator_tokenizer(prompt, return_tensors="pt", max_length=128, truncation=True)
-        outputs = generator.generate(
-            inputs["input_ids"],
-            max_length=char_limit // 2,  # Keep it concise
-            num_return_sequences=2,
-            temperature=0.9,
-            top_p=0.85,
-            do_sample=True,
-            pad_token_id=generator_tokenizer.eos_token_id,
-            no_repeat_ngram_size=2,
-            min_length=30,
-            repetition_penalty=1.2
-        )
+    # Create marketing message templates based on tone
+    intro_phrases = {
+        'professional': [
+            f"Introducing {product_name}:",
+            f"Discover {product_name}:",
+            f"Meet {product_name}:",
+        ],
+        'casual': [
+            f"Check out {product_name}!",
+            f"Say hello to {product_name}!",
+            f"Get ready for {product_name}!",
+        ],
+        'friendly': [
+            f"You'll love {product_name}!",
+            f"Experience {product_name}!",
+            f"We're excited to present {product_name}!",
+        ]
+    }
+    
+    # Use professional tone as default if tone not found
+    selected_tone = tone.lower() if tone.lower() in intro_phrases else 'professional'
+    
+    import random
+    
+    def create_post():
+        # Start with a random intro
+        post = random.choice(intro_phrases[selected_tone])
+        
+        # Add main description
+        post += f" {product_description}"
+        
+        # Add a key feature and benefit
+        feature = random.choice(features)
+        benefit = random.choice(benefits)
+        
+        if len(post) + len(feature) + len(benefit) + 10 < char_limit:
+            post += f" Featuring {feature}."
+            post += f" {benefit}."
         
-        generated_texts = [generator_tokenizer.decode(output, skip_special_tokens=True) for output in outputs]
+        # Add target audience if space allows
+        if len(post) + len(target_audience) + 15 < char_limit:
+            post += f" Perfect for {target_audience}."
         
-        # Process and clean the generated content
+        # Add call to action
+        if platform == "Twitter":
+            if len(post) + 30 < char_limit:
+                post += " Learn more at our website!"
+        else:  # Instagram
+            if len(post) + 20 < char_limit:
+                post += f"\n\n#{product_name.replace(' ', '')}"
+        
+        return post.strip()
+    
+    try:
+        # Generate multiple versions
+        posts = [create_post() for _ in range(2)]
         filtered_content = []
-        for text in generated_texts:
-            # Extract only the part after "Marketing post in {tone} tone:"
-            try:
-                post = text.split("Marketing post in")[-1].split("tone:")[-1].strip()
-            except:
-                post = text
-            
-            # Basic cleaning
-            post = ' '.join(post.split())  # Remove extra whitespace
-            
-            # Skip if the post is too short or contains prompt artifacts
-            if len(post) < 20 or "Marketing post" in post or "tone:" in post:
-                continue
-                
-            # Ensure it starts with product name if it's not already included
-            if product_name not in post:
-                post = f"{product_name}: {post}"
-            
-            # Ensure there's a call to action
-            if "learn more" not in post.lower() and len(post) + 15 <= char_limit:
-                post += " Learn more today!"
+        
+        for post in posts:
+            # Verify length
+            if len(post) > char_limit:
+                post = post[:char_limit-3] + "..."
             
             # Check sentiment and safety
             try:
@@ -127,7 +144,7 @@ Marketing post in {tone} tone:
                 safety_check = content_checker(post)[0]
                 
                 filtered_content.append({
-                    'text': post[:char_limit],
+                    'text': post,
                     'sentiment': sentiment['label'],
                     'safety_score': f"{float(safety_check.get('score', 0)):.2f}"
                 })
@@ -135,25 +152,16 @@ Marketing post in {tone} tone:
                 print(f"Error in content analysis: {str(e)}")
                 continue
         
-        # If no valid content was generated, use a structured fallback
-        if not filtered_content:
-            benefit = unique_benefits.split(',')[0].strip()
-            feature = key_features.split(',')[0].strip()
-            
-            fallback_post = f"{product_name}: {product_description} {benefit}. Featuring {feature}. Perfect for {target_audience}. Learn more today!"
-            
-            filtered_content.append({
-                'text': fallback_post[:char_limit],
-                'sentiment': 'positive',
-                'safety_score': '1.00'
-            })
-        
-        return filtered_content
-
+        return filtered_content if filtered_content else [{
+            'text': create_post(),
+            'sentiment': 'positive',
+            'safety_score': '1.00'
+        }]
+    
     except Exception as e:
         print(f"Error in content generation: {str(e)}")
         return [{
-            'text': f"{product_name}: {product_description[:100]}... Learn more today!",
+            'text': f"Introducing {product_name}: {product_description[:100]}... Learn more!",
             'sentiment': 'neutral',
             'safety_score': '1.00'
         }]