Update app.py

app.py

@@ -2,8 +2,12 @@ import gradio as gr
 from transformers import pipeline
 import time
 
-# 加载小一点的模型以提高速度
-model = pipeline("text2text-generation", model="google/flan-t5-base")
+# 尝试更强的模型（如 flan-t5-large）；如果内存不够，可改回 base
+model = pipeline(
+    "text2text-generation",
+    model="google/flan-t5-large",
+    device_map="auto"
+)
 
 def analyze_text(text):
     """分析英文原著文本，每句分析时显示进度。"""
@@ -18,19 +22,33 @@ def analyze_text(text):
 
     all_results = []
     for i, sentence in enumerate(sentences, start=1):
-        # 生成prompt
-        prompt = f"Analyze the following English sentence for vocabulary, grammar, and meaning:\n\n{sentence}\n\nGive a short analysis in English."
-        
-        # 调用模型
-        result = model(prompt, max_length=150)[0]['generated_text']
-        all_results.append(f"Sentence {i}: {result}")
-        
-        # 每次yield一个更新，显示进度
+        prompt = f"""
+You are an advanced English literature analysis assistant.
+Please analyze the following sentence from a literary perspective.
+Explain:
+1. Grammar and sentence structure
+2. Vocabulary richness
+3. Idiomatic/natural usage
+4. Possible literary meaning or tone
+Then give a short summary (in English).
+
+Sentence:
+"{sentence}"
+"""
+        result = model(
+            prompt,
+            max_length=512,
+            do_sample=True,
+            temperature=0.7
+        )[0]['generated_text']
+
+        all_results.append(f"Sentence {i}:\n{result}\n")
+
         progress = int(i / total * 100)
-        yield f"⏳ 分析进度：{i}/{total} ({progress}%)\n\n" + "\n\n".join(all_results)
-        time.sleep(0.3)  # 轻微延迟让前端看到变化
+        yield f"⏳ 分析进度：{i}/{total} ({progress}%)\n\n" + "\n".join(all_results)
+        time.sleep(0.5)
 
-    yield f"✅ 分析完成！共 {total} 句。\n\n" + "\n\n".join(all_results)
+    yield f"✅ 分析完成！共 {total} 句。\n\n" + "\n".join(all_results)
 
 
 # 界面
@@ -38,9 +56,8 @@ demo = gr.Interface(
     fn=analyze_text,
     inputs=gr.Textbox(label="输入英文原著片段", lines=10, placeholder="例如：It was the best of times, it was the worst of times..."),
     outputs=gr.Textbox(label="分析结果", lines=15),
-    title="📚 英文原著阅读助手",
-    description="自动逐句分析英文原著中的词汇、语法和含义。支持进度显示。",
-    live=False
+    title="📚 英文原著阅读与分析助手",
+    description="逐句分析英文原著的语法、词汇和文体特征，并实时显示进度。",
 )
 
 if __name__ == "__main__":