Update app.py

app.py

@@ -11,25 +11,40 @@ class TwitterCloneApp:
         self.processor = None
         
     def process_upload(self, file):
-        """Process uploaded PDF file and analyze personality"""
+        """Process uploaded PDF file and analyze personality."""
         try:
+            if not file:
+                return "Error: No file uploaded. Please upload a PDF dataset."
+            
             self.processor = TweetDatasetProcessor()
             text = self.processor.extract_text_from_pdf(file.name)
             df = self.processor.process_pdf_content(text)
-            
+
+            # Extract mentions and hashtags
             mentions = df['mentions'].explode().dropna().unique().tolist()
             hashtags = df['hashtags'].explode().dropna().unique().tolist()
 
+            # Perform personality analysis
             personality_analysis = self.processor.analyze_personality()
             
-            return f"Processed {len(df)} tweets\n\nMentions: {mentions}\n\nHashtags: {hashtags}\n\nPersonality Analysis:\n{personality_analysis}"
+            # Format output
+            result = f"""
+### Analysis Complete
+- **Processed Tweets**: {len(df)}
+- **Mentions**: {", ".join(mentions) if mentions else "None"}
+- **Hashtags**: {", ".join(hashtags) if hashtags else "None"}
+
+### Personality Analysis
+{personality_analysis}
+"""
+            return result
         except Exception as e:
             return f"Error processing file: {str(e)}"
     
     def generate_tweet(self, context):
-        """Generate a new tweet based on the analyzed personality"""
+        """Generate a new tweet based on the analyzed personality."""
         if not self.processor:
-            return "Please upload and analyze a dataset first."
+            return "Error: Please upload and analyze a dataset first."
         
         try:
             # Predefined contexts
@@ -48,18 +63,18 @@ class TwitterCloneApp:
             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
+            # Prioritize user context if provided
             if context:
-                selected_contexts.append(context)
+                selected_contexts.insert(0, context)
 
-            # Generate the tweet with diverse contexts
+            # Generate the tweet
             tweet = self.processor.generate_tweet(context=" | ".join(selected_contexts))
-            return tweet
+            return f"### Generated Tweet\n{tweet}"
         except Exception as e:
             return f"Error generating tweet: {str(e)}"
     
     def create_interface(self):
-        """Create the Gradio interface"""
+        """Create the Gradio interface."""
         with gr.Blocks(title="Twitter Personality Cloner") as interface:
             gr.Markdown("# Twitter Personality Cloner")
             gr.Markdown("Upload a PDF file containing tweets to analyze the author's personality and generate new tweets in their style.")
@@ -67,7 +82,7 @@ class TwitterCloneApp:
             with gr.Tab("Analyze Personality"):
                 file_input = gr.File(label="Upload PDF Dataset", file_types=[".pdf"])
                 analyze_button = gr.Button("Analyze Dataset")
-                analysis_output = gr.Textbox(label="Analysis Results", lines=10)
+                analysis_output = gr.Textbox(label="Analysis Results", lines=10, interactive=False)
                 
                 analyze_button.click(
                     fn=self.process_upload,
@@ -78,7 +93,7 @@ class TwitterCloneApp:
             with gr.Tab("Generate Tweets"):
                 context_input = gr.Textbox(label="Context (optional)", placeholder="Enter topic or context for the tweet")
                 generate_button = gr.Button("Generate Tweet")
-                tweet_output = gr.Textbox(label="Generated Tweet")
+                tweet_output = gr.Textbox(label="Generated Tweet", lines=3, interactive=False)
                 
                 generate_button.click(
                     fn=self.generate_tweet,
@@ -97,3 +112,4 @@ if __name__ == "__main__":
     main()
 
 
+