Update llm_utils.py

llm_utils.py

@@ -102,23 +102,29 @@ def analyze_meeting_transcript(transcript_text):
             
     return industry, meeting_purpose, suggested_template
 
-def generate_summary(transcript_text, reference_summary_text=None, auto_template_structure=None):
-    """Generates the meeting summary."""
+def generate_summary(transcript_text, reference_summary_text=None, auto_template_structure=None, user_notes=None):
+    """Generates the meeting summary with optional user notes."""
+    # Process user notes - if empty, provide appropriate message
+    processed_user_notes = "无用户手动记录的笔记" if not user_notes or not user_notes.strip() else user_notes.strip()
+    
     if reference_summary_text:
         prompt = SUMMARIZE_WITH_REFERENCE_TEMPLATE_PROMPT.format(
             transcript_text=transcript_text,
-            reference_summary_text=reference_summary_text
+            reference_summary_text=reference_summary_text,
+            user_notes=processed_user_notes
         )
     elif auto_template_structure:
         prompt = SUMMARIZE_WITH_AUTO_TEMPLATE_PROMPT.format(
             transcript_text=transcript_text,
-            template_structure=auto_template_structure
+            template_structure=auto_template_structure,
+            user_notes=processed_user_notes
         )
     else:
         # Fallback or default summarization if no template is provided
         # This case should ideally be handled by ensuring one of the templates is always available
-        prompt = f"Please summarize the following meeting transcript:\n\n{transcript_text}"
-        
+        user_notes_text = f"\n\nUser's Manual Notes:\n{user_notes}" if user_notes and user_notes.strip() else ""
+        prompt = f"Please summarize the following meeting transcript:{user_notes_text}\n\n{transcript_text}"
+    #print(f"Prompt: {prompt}")
     return call_openrouter_api(prompt)
 
 if __name__ == '__main__':
@@ -143,7 +149,7 @@ if __name__ == '__main__':
             print(f"Suggested Template:\n{template}")
             
             print("\n--- Generating Summary (with auto template) ---")
-            summary1 = generate_summary(transcript, auto_template_structure=template)
+            summary1 = generate_summary(transcript, auto_template_structure=template, user_notes="测试关键笔记：重要决策点")
             print(f"Summary 1:\n{summary1}")
             
             print("\n--- Generating Summary (with example reference template) ---")
@@ -159,7 +165,7 @@ if __name__ == '__main__':
             **Action Items:**
             - User A: Task X (Due: YYYY-MM-DD)
             """
-            summary2 = generate_summary(transcript, reference_summary_text=example_ref_summary)
+            summary2 = generate_summary(transcript, reference_summary_text=example_ref_summary, user_notes="测试关键笔记：关注预算和时间节点")
             print(f"Summary 2:\n{summary2}")
         else:
             print(f"Could not parse {example_srt_path} for testing.")