Update app.py

app.py

@@ -1,5 +1,6 @@
 import gradio as gr
 import os
+import random
 from tweet_analyzer import TweetDatasetProcessor
 from dotenv import load_dotenv
 
@@ -31,7 +32,28 @@ class TwitterCloneApp:
             return "Please upload and analyze a dataset first."
         
         try:
-            tweet = self.processor.generate_tweet(context)
+            # Predefined contexts
+            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."
+            ]
+
+            # Extract historical topics
+            historical_topics = self.processor.analyze_topics(n_topics=5)
+
+            # Combine predefined contexts with historical topics
+            combined_contexts = additional_contexts + historical_topics
+            selected_contexts = random.sample(combined_contexts, min(3, len(combined_contexts)))
+
+            # If user provided context, add it to the mix
+            if context:
+                selected_contexts.append(context)
+
+            # Generate the tweet with diverse contexts
+            tweet = self.processor.generate_tweet(context=" | ".join(selected_contexts))
             return tweet
         except Exception as e:
             return f"Error generating tweet: {str(e)}"
@@ -74,3 +96,4 @@ def main():
 if __name__ == "__main__":
     main()
 
+