Update tweet_analyzer.py

tweet_analyzer.py

@@ -109,22 +109,33 @@ class TweetDatasetProcessor:
 
     def generate_tweet(self, context=""):
         """Generate a new tweet based on personality profile and optional context"""
+        additional_contexts = [
+            "Comment on a recent technological advancement.",
+            "Share a motivational thought.",
+            "Discuss a current trending topic.",
+            "Reflect on a past experience.",
+            "Provide advice to followers."
+        ]
+    
+        # Randomly choose an additional context to diversify tweets
+        import random
+        random_context = random.choice(additional_contexts)
+    
         generation_prompt = f"""Based on this personality profile:
         {self.personality_profile}
-
         Current context or topic (if any):
         {context}
-
-        Generate a tweet that this person would write right now. Consider:
-        1. Their core beliefs and values
-        2. Their typical emotional expression
-        3. Their communication style and vocabulary
-        4. Their knowledge areas and experiences
-        5. Current context (if provided)
-
-        The tweet should feel indistinguishable from their authentic tweets.
+        Additionally, consider this specific context:
+       {random_context}
+       Generate a tweet that this person would write right now. Consider:
+       1. Their core beliefs and values
+       2. Their typical emotional expression
+       3. Their communication style and vocabulary
+       4. Their knowledge areas and experiences
+       5. Current context (if provided)
+       The tweet should feel indistinguishable from their authentic tweets.
         """
-        
+    
         response = self.groq_client.chat.completions.create(
             messages=[
                 {
@@ -137,6 +148,8 @@ class TweetDatasetProcessor:
                 }
             ],
             model="mixtral-8x7b-32768",
-            temperature=0.7,
+            temperature=0.8,  # Increase temperature for more diversity
+            max_tokens=150,   # Adjust max tokens for more substantial tweets
         )
-        return response.choices[0].message.content
+    
+        return response.choices[0].message.content