Update app.py

app.py

@@ -165,20 +165,31 @@ def extract_information(api_key, prompt, examples_json, input_text, template_nam
     try:
         os.environ['LANGEXTRACT_API_KEY'] = api_key
         
-        result = lx.extract(
+        results = lx.extract(
             text_or_documents=input_text,
             prompt_description=prompt,
             examples=examples,
             model_id="gemini-2.5-flash",
         )
         
+        # langextract 返回的是列表，取第一个结果
+        if isinstance(results, list) and len(results) > 0:
+            result = results[0]
+        else:
+            result = results
+        
+        # 构建输出字典
         output_dict = {
-            "source_text": result.source_text,
-            "extractions": [ext.to_dict() for ext in result.extractions]
+            "source_text": result.text if hasattr(result, 'text') else input_text,
+            "extractions": [ext.to_dict() for ext in result.extractions] if hasattr(result, 'extractions') else []
         }
 
+        # 保存文件
         with tempfile.NamedTemporaryFile(mode='w+', delete=False, suffix='.jsonl', encoding='utf-8') as tmp_file:
-            lx.io.save_annotated_documents([result], file_path=tmp_file.name)
+            if isinstance(results, list):
+                lx.io.save_annotated_documents(results, file_path=tmp_file.name)
+            else:
+                lx.io.save_annotated_documents([results], file_path=tmp_file.name)
             download_path = tmp_file.name
         
         stats = generate_statistics(output_dict)
@@ -187,6 +198,9 @@ def extract_information(api_key, prompt, examples_json, input_text, template_nam
         return output_dict, download_path, stats, history, gr.update(visible=True)
 
     except Exception as e:
+        import traceback
+        error_detail = traceback.format_exc()
+        print(f"详细错误信息:\n{error_detail}")
         raise gr.Error(f"❌ 提取失败: {str(e)}")
 
 def load_template(template_name):