Update app.py

app.py

@@ -10,94 +10,86 @@ API_KEY = os.getenv("API_KEY")
 HF_MODEL_URL = "https://api-inference.huggingface.co/models/tabularisai/multilingual-sentiment-analysis"
 
 
-def analyze_post_gradio(post):
-    """
-    Gradio version of the analyze_post function
-    """
+def analyze_post_gradio(post: str):
+    """Gradio version of the analyze_post function."""
     try:
+        if not post or not post.strip():
+            return "⚠️ Please enter a valid post before analyzing."
+
         # Clean the text
         cleaned_post = clean_text(post)
-        
+
         # Get sentiment analysis
         sentiment_result = query_hf_sentiment(cleaned_post, API_KEY, HF_MODEL_URL)
-        sentiment = sentiment_result['raw_output'][0]
-        
+
+        # Handle possible formats of HF output
+        if isinstance(sentiment_result, dict) and "raw_output" in sentiment_result:
+            sentiment = sentiment_result["raw_output"][0]
+        elif isinstance(sentiment_result, list) and len(sentiment_result) > 0:
+            sentiment = sentiment_result[0]
+        else:
+            sentiment = {"label": "Unknown", "score": 0.0}
+
         # Extract themes
-        themes = extract_themes(cleaned_post, create_openai_client(API_KEY))
-        
-        # Prepare data for storage and display
-        data = {
-            "post": post,
-            "cleaned_post": cleaned_post,
-            "sentiment": {
-                "label": sentiment['label'],
-                "score": sentiment['score']
-            },
-            "theme": themes
-        }
-        
-        # Store in MongoDB
-        #insert_post(data)
-        
-        # Format output for display
+        openai_client = create_openai_client(API_KEY)
+        themes = extract_themes(cleaned_post, openai_client)
+
+        # Prepare data
         output = f"""
-        **Original Post:** {post}
-        
-        **Cleaned Post:** {cleaned_post}
-        
-        **Sentiment Analysis:**
-        - Label: {sentiment['label']}
-        - Score: {sentiment['score']:.4f}
-        
-        **Extracted Themes:** {themes}
-        
-        **Status:** ✅ Post analyzed and stored successfully in MongoDB
-        """
-        
+### 🧾 **Original Post**
+{post}
+
+### 🧹 **Cleaned Post**
+{cleaned_post}
+
+### 💬 **Sentiment Analysis**
+- **Label:** {sentiment.get('label', 'Unknown')}
+- **Score:** {sentiment.get('score', 0.0):.4f}
+
+### 🏷️ **Extracted Themes**
+{themes}
+
+✅ **Status:** Post analyzed successfully.
+"""
         return output
-        
+
     except Exception as e:
-        return f"❌ Error analyzing post: {str(e)}"
+        return f"❌ **Error analyzing post:** {str(e)}"
 
-# Create Gradio interface
+
+# --- Gradio Interface ---
 with gr.Blocks(title="Post Analyzer", theme=gr.themes.Soft()) as demo:
     gr.Markdown("# 📊 Post Analysis Tool")
-    gr.Markdown("Enter a post to analyze its sentiment and extract themes")
-    
+    gr.Markdown("Enter a post to analyze its sentiment and extract key themes using AI models.")
+
     with gr.Row():
         with gr.Column():
             post_input = gr.Textbox(
                 label="Enter your post",
                 placeholder="Type your post here...",
                 lines=4,
-                max_lines=10
+                max_lines=10,
             )
-            analyze_btn = gr.Button("Analyze Post", variant="primary")
-        
+            analyze_btn = gr.Button("🔍 Analyze Post", variant="primary")
+
         with gr.Column():
-            output = gr.Markdown(label="Analysis Results")
-    
-    # Set up the interaction
+            output = gr.Markdown("Results will appear here after analysis...")
+
     analyze_btn.click(
         fn=analyze_post_gradio,
         inputs=post_input,
-        outputs=output
+        outputs=output,
     )
-    
-    # Add examples
+
     gr.Examples(
         examples=[
             "I absolutely love this product! It's amazing and works perfectly.",
             "I'm really disappointed with the service. It was slow and unhelpful.",
-            "The weather today is nice, but I'm concerned about climate change."
+            "The weather today is nice, but I'm concerned about climate change.",
         ],
-        inputs=post_input
+        inputs=post_input,
     )
 
-# Launch the app
+# --- Launch for Hugging Face Space ---
 if __name__ == "__main__":
-    demo.launch(
-        server_name="0.0.0.0",
-        server_port=7860,
-        share=True
-    )
+    demo.launch()