small changes

app.py

@@ -5,6 +5,8 @@ import nltk
 from datetime import datetime, timedelta
 import requests
 from bs4 import BeautifulSoup
+import time
+import emoji
 
 # Download required NLTK data
 try:
@@ -90,7 +92,6 @@ def generate_content(
             f"Check out {product_name}!",
             f"Say hello to {product_name}!",
             f"Get ready for {product_name}!",
-            f"You'll love {product_name}!",
             f"Looking for something special? Try {product_name}!",
             f"Meet your new favorite: {product_name}!",
             f"Game-changer alert: {product_name} is here!",
@@ -102,7 +103,6 @@ def generate_content(
             f"Ready to discover {product_name}?",
             f"Here's why you'll love {product_name}:",
             f"Make every day better with {product_name}!",
-            f"Transform your experience with {product_name}!",
         ]
     }
 
@@ -157,7 +157,6 @@ def generate_content(
             "Visit our website!",
             "Join us today!",
             "Transform your life today!",
-            "Start your journey!",
         ],
         'Instagram': [
             f"\n\n#{product_name.replace(' ', '')}",
@@ -268,8 +267,75 @@ def generate_content(
             'safety_score': '1.00'
         }]
 
+def process_input_with_loading(
+    product_name,
+    product_description,
+    target_audience,
+    key_features,
+    unique_benefits,
+    platform,
+    tone,
+    progress=gr.Progress()
+):
+    # Initial loading message
+    features_list = """⚡ While I generate your content, here's what I can do:
+    
+📝 Generate multiple unique marketing messages
+🎯 Adapt content for different platforms
+🔍 Ensure ethical content generation
+📊 Analyze sentiment and safety
+✨ Create platform-specific formatting
+    
+Processing your request..."""
+    
+    yield features_list + "\n\n⏳ Starting generation..."
+    time.sleep(1)
+    
+    # Update message with steps
+    steps = [
+        "Loading language models...",
+        "Analyzing product information...",
+        "Generating content variations...",
+        "Checking content safety...",
+        "Finalizing output..."
+    ]
+    
+    for i, step in enumerate(steps, 1):
+        progress(i/len(steps))
+        yield features_list + f"\n\n⏳ {step}"
+        time.sleep(1)
+    
+    # Generate actual content
+    try:
+        results = generate_content(
+            product_name,
+            product_description,
+            target_audience,
+            key_features,
+            unique_benefits,
+            platform,
+            tone,
+            generator_tokenizer,
+            generator,
+            sentiment_analyzer,
+            content_checker
+        )
+        
+        output = "🎯 Generated Marketing Content:\n\n"
+        for i, content in enumerate(results, 1):
+            output += f"Version {i}:\n"
+            output += f"📝 Content: {content['text']}\n"
+            output += f"😊 Sentiment: {content['sentiment']}\n"
+            output += f"✅ Safety Score: {content['safety_score']}\n"
+            output += "-" * 50 + "\n"
+        
+        yield output
+    except Exception as e:
+        yield f"An error occurred: {str(e)}"
+
 def create_interface():
     print("Loading models...")
+    global generator_tokenizer, generator, sentiment_analyzer, content_checker
     generator_tokenizer, generator, sentiment_analyzer, content_checker = load_models()
     print("Models loaded successfully!")
     
@@ -287,74 +353,6 @@ def create_interface():
     def clear_form():
         return [""] * 7  # Returns empty strings for all 7 input fields
 
-    import time
-    
-    def process_input_with_loading(
-        product_name,
-        product_description,
-        target_audience,
-        key_features,
-        unique_benefits,
-        platform,
-        tone,
-        progress=gr.Progress()
-    ):
-        # Initial loading message
-        features_list = """⚡ While I generate your content, here's what I can do:
-        
-📝 Generate multiple unique marketing messages
-🎯 Adapt content for different platforms
-🔍 Ensure ethical content generation
-📊 Analyze sentiment and safety
-✨ Create platform-specific formatting
-        
-Processing your request..."""
-        
-        yield features_list + "\n\n⏳ Starting generation..."
-        time.sleep(1)
-        
-        # Update message with steps
-        steps = [
-            "Loading language models...",
-            "Analyzing product information...",
-            "Generating content variations...",
-            "Checking content safety...",
-            "Finalizing output..."
-        ]
-        
-        for i, step in enumerate(steps, 1):
-            progress(i/len(steps))
-            yield features_list + f"\n\n⏳ {step}"
-            time.sleep(1)
-        
-        # Generate actual content
-        try:
-            results = generate_content(
-                product_name,
-                product_description,
-                target_audience,
-                key_features,
-                unique_benefits,
-                platform,
-                tone,
-                generator_tokenizer,
-                generator,
-                sentiment_analyzer,
-                content_checker
-            )
-            
-            output = "🎯 Generated Marketing Content:\n\n"
-            for i, content in enumerate(results, 1):
-                output += f"Version {i}:\n"
-                output += f"📝 Content: {content['text']}\n"
-                output += f"😊 Sentiment: {content['sentiment']}\n"
-                output += f"✅ Safety Score: {content['safety_score']}\n"
-                output += "-" * 50 + "\n"
-            
-            yield output
-        except Exception as e:
-            yield f"An error occurred: {str(e)}"
-
     # Create the interface with blocks for custom layout
     with gr.Blocks(theme=gr.themes.Default()) as demo:
         gr.Markdown("# Ethimar - AI Marketing Content Generator")
@@ -366,7 +364,7 @@ Processing your request..."""
                 "Fill the form with sample data",
                 variant="primary",
                 size="sm",
-                scale=0.2
+                scale=1
             )
 
         # Main content area with two columns

requirements.txt

@@ -4,4 +4,5 @@ transformers
 nltk
 requests
 beautifulsoup4
-lxml
+lxml
+emoji==0.6.0