Refactor app to implement multilingual sentiment analysis using Hugging Face's pipeline, replacing the previous prompt enhancement functionality. Update Gradio interface and adjust requirements for dependencies.

app.py

@@ -1,29 +1,42 @@
-import os
 import gradio as gr
-from huggingface_hub import InferenceClient
+from transformers import pipeline
 
-# 使用 Space Secrets 管理 HF_TOKEN，而不是明文写在代码里
-HF_TOKEN = os.environ.get("HF_TOKEN", "your_hf_token_here")
+# 直接引用 Hugging Face Hub 上的模型
+MODEL_REPO = "tabularisai/multilingual-sentiment-analysis"
 
-def enhance_prompt(prompt):
-    client = InferenceClient(token=HF_TOKEN, model="PromptEnhancer/PromptEnhancer-Img2img-Edit")
+# 初始化情感分析 pipeline
+classifier = pipeline("sentiment-analysis", model=MODEL_REPO)
 
+def analyze_sentiment(text):
     try:
-        # 使用 text_generation 方法而不是直接调用 client()
-        response = client.text_generation(prompt)
-        # 有些模型返回可能在 'generated_text' 或 'enhanced_prompt' 字段里
-        return response.get("enhanced_prompt") or response.get("generated_text") or "No enhanced prompt returned."
+        results = classifier(text)
+        # 处理单条和多条输入
+        if isinstance(results, list):
+            output_lines = []
+            for result in results:
+                label = result.get('label', '')
+                score = result.get('score', 0)
+                output_lines.append(f"Label: {label}, Confidence: {score:.2%}")
+            return "\n".join(output_lines)
+        else:
+            label = results.get('label', '')
+            score = results.get('score', 0)
+            return f"Label: {label}, Confidence: {score:.2%}"
     except Exception as e:
         return f"Error: {str(e)}"
 
-with gr.Blocks(title="Prompt Enhancer") as app:
-    gr.Markdown("## ✨ Prompt Enhancer\nEnhance your prompts using the Hugging Face Prompt Enhancer model.")
+# Gradio Web 界面
+with gr.Blocks(title="Multilingual Sentiment Analysis") as app:
+    gr.Markdown("## 🌍 Multilingual Sentiment Analysis\nAnalyze sentiment for texts in multiple languages using [tabularisai/multilingual-sentiment-analysis](https://huggingface.co/tabularisai/multilingual-sentiment-analysis).")
 
-    prompt_input = gr.Textbox(label="Enter your prompt here")
-    output = gr.Textbox(label="Enhanced Prompt")
-
-    btn = gr.Button("Enhance Prompt")
-    btn.click(fn=enhance_prompt, inputs=prompt_input, outputs=output)
+    prompt_input = gr.Textbox(label="Enter your text here", lines=3, placeholder="Type or paste your sentence...")
+    output = gr.Textbox(label="Sentiment Result", lines=2)
+    btn = gr.Button("Analyze Sentiment")
+    btn.click(
+        fn=analyze_sentiment,
+        inputs=prompt_input,
+        outputs=output
+    )
 
 if __name__ == "__main__":
-    app.launch()
+    app.launch()

requirements.txt

@@ -2,5 +2,5 @@ torch
 transformers
 Pillow
 
-gradio>=4.44.0
-huggingface_hub>=0.24.6
+gradio
+huggingface_hub