Update main.py

main.py

@@ -336,16 +336,16 @@ async def _call_gemini(text: str, summary_type: str) -> Tuple[Optional[str], Opt
         return None, "Error: Primary AI service (Gemini) not configured/available."
     logger.info(f"[Gemini Primary] Generating {summary_type} summary using {GEMINI_MODEL}. Input length: {len(text)}")
 
-    # Define prompts (same logic as before)
+    # Define prompts
     if summary_type == "paragraph": prompt = ("You are an AI model designed to provide concise summaries using British English spellings. Your output MUST be:\n" "• Clear and simple language suitable for someone unfamiliar with the topic.\n" "• Uses British English spellings throughout.\n" "• Straightforward and understandable vocabulary; avoid complex terms.\n" "• Presented as ONE SINGLE PARAGRAPH.\n" "• No more than 85 words maximum; but does not have to be exactly 85.\n" "• Considers the entire text content equally.\n" "• Uses semicolons (;) instead of em dashes (– or —).\n\n" "Here is the text to summarise:")
     else: # points summary
-        prompt = ("You are an AI model designed to provide concise summaries using British English spellings. Your output MUST strictly follow this Markdown format:\n\n"
+        prompt = ("You are an AI model designed to provide concise summaries using British English spellings. Your output MUST strictly follow this format:\n\n"
                   "• For each distinct topic or section identified in the text, create a heading.\n"
-                  "• Each heading MUST be enclosed in double asterisks for bolding (e.g., **Section Title**).\n"
+                  "• Each heading MUST be plain text without any formatting (e.g., Section Title).\n" # MODIFIED HERE
                   "• Immediately following each heading, list the key points as a bulleted list.\n"
                   "• Each bullet point MUST start with a hyphen and a space (- ) on a new line.\n"
                   "• The text within each bullet point should NOT contain any bold formatting.\n"
-                  "• IMPORTANT: Never use bold formatting (double asterisks) within the text of the bullet points themselves.\n" # Added instruction for emphasis
+                  "• IMPORTANT: Never use bold formatting (double asterisks) within the text of the bullet points themselves.\n"
                   "• Use clear, simple, and straightforward language suitable for someone unfamiliar with the topic.\n"
                   "• Use British English spellings throughout.\n"
                   "• Avoid overly complex or advanced vocabulary.\n"
@@ -401,9 +401,10 @@ async def _call_gemini(text: str, summary_type: str) -> Tuple[Optional[str], Opt
 
         if summary:
             logger.info(f"[Gemini Primary] Success generating summary. Output len: {len(summary)}");
-            # Escape Markdown for Telegram
-            escaped_summary = summary.strip().replace('_', r'\_').replace('*', r'\*').replace('[', r'\[').replace('`', r'\`')
-            return escaped_summary, None
+            # Escape Markdown for Telegram only if necessary (Removed escaping as it might conflict with plain heading)
+            # escaped_summary = summary.strip().replace('_', r'\_').replace('*', r'\*').replace('[', r'\[').replace('`', r'\`')
+            # Let's see if the AI respects the instruction without further escaping. If not, we might need selective escaping.
+            return summary.strip(), None
         else:
             finish_reason = response.candidates[0].finish_reason if response.candidates else 'N/A'
             logger.warning(f"[Gemini Primary] Gemini returned empty summary or content was blocked. Finish reason: {finish_reason}");
@@ -421,16 +422,16 @@ async def _call_openrouter(text: str, summary_type: str) -> Tuple[Optional[str],
         return None, "Error: Fallback AI service (OpenRouter) not configured/available."
     logger.info(f"[OpenRouter Fallback] Generating {summary_type} summary using {OPENROUTER_MODEL}. Input length: {len(text)}")
 
-    # Define prompts (same logic as before)
+    # Define prompts
     if summary_type == "paragraph": prompt = ("You are an AI model designed to provide concise summaries using British English spellings. Your output MUST be:\n" "• Clear and simple language suitable for someone unfamiliar with the topic.\n" "• Uses British English spellings throughout.\n" "• Straightforward and understandable vocabulary; avoid complex terms.\n" "• Presented as ONE SINGLE PARAGRAPH.\n" "• No more than 85 words maximum; but does not have to be exactly 85.\n" "• Considers the entire text content equally.\n" "• Uses semicolons (;) instead of em dashes (– or —).\n\n" "Here is the text to summarise:")
     else: # points summary
-        prompt = ("You are an AI model designed to provide concise summaries using British English spellings. Your output MUST strictly follow this Markdown format:\n\n"
+        prompt = ("You are an AI model designed to provide concise summaries using British English spellings. Your output MUST strictly follow this format:\n\n"
                   "• For each distinct topic or section identified in the text, create a heading.\n"
-                  "• Each heading MUST be enclosed in double asterisks for bolding (e.g., **Section Title**).\n"
+                  "• Each heading MUST be plain text without any formatting (e.g., Section Title).\n" # MODIFIED HERE
                   "• Immediately following each heading, list the key points as a bulleted list.\n"
                   "• Each bullet point MUST start with a hyphen and a space (- ) on a new line.\n"
                   "• The text within each bullet point should NOT contain any bold formatting.\n"
-                  "• IMPORTANT: Never use bold formatting (double asterisks) within the text of the bullet points themselves.\n" # Added instruction for emphasis
+                  "• IMPORTANT: Never use bold formatting (double asterisks) within the text of the bullet points themselves.\n"
                   "• Use clear, simple, and straightforward language suitable for someone unfamiliar with the topic.\n"
                   "• Use British English spellings throughout.\n"
                   "• Avoid overly complex or advanced vocabulary.\n"
@@ -470,9 +471,9 @@ async def _call_openrouter(text: str, summary_type: str) -> Tuple[Optional[str],
                             summary = message.get("content")
                             if summary:
                                 logger.info(f"[OpenRouter Fallback] Success via OpenRouter. Output len: {len(summary)}")
-                                # Escape Markdown for Telegram
-                                escaped_summary = summary.strip().replace('_', r'\_').replace('*', r'\*').replace('[', r'\[').replace('`', r'\`')
-                                return escaped_summary, None
+                                # Escape Markdown for Telegram only if necessary (Removed escaping)
+                                # escaped_summary = summary.strip().replace('_', r'\_').replace('*', r'\*').replace('[', r'\[').replace('`', r'\`')
+                                return summary.strip(), None
                             else:
                                 logger.warning(f"[OpenRouter Fallback] OpenRouter success but content empty. Resp: {data}")
                                 return None, "Sorry, the fallback AI model returned an empty summary."
@@ -513,6 +514,7 @@ async def _call_openrouter(text: str, summary_type: str) -> Tuple[Optional[str],
 async def generate_summary(text: str, summary_type: str) -> str:
     """
     Generates summary using Gemini (Primary) and falls back to OpenRouter if needed.
+    Ensures point-based summaries have plain text headings.
     """
     global _gemini_primary_enabled, _openrouter_fallback_enabled, GEMINI_MODEL, OPENROUTER_MODEL
     logger.info(f"[Summary Generation] Starting process. Primary: Gemini ({GEMINI_MODEL}), Fallback: OpenRouter ({OPENROUTER_MODEL})")
@@ -526,6 +528,7 @@ async def generate_summary(text: str, summary_type: str) -> str:
         final_summary, error_message = await _call_gemini(text, summary_type)
         if final_summary:
             logger.info(f"[Summary Generation] Success with primary AI (Gemini).")
+            # Return directly - prompt handles formatting
             return final_summary
         else:
             logger.warning(f"[Summary Generation] Primary AI (Gemini) failed or returned unusable result. Error: {error_message}. Proceeding to fallback.")
@@ -539,6 +542,7 @@ async def generate_summary(text: str, summary_type: str) -> str:
         fallback_summary, fallback_error = await _call_openrouter(text, summary_type)
         if fallback_summary:
             logger.info(f"[Summary Generation] Success with fallback AI (OpenRouter).")
+            # Return directly - prompt handles formatting
             return fallback_summary
         else:
             logger.error(f"[Summary Generation] Fallback AI (OpenRouter) also failed. Error: {fallback_error}")
@@ -602,8 +606,10 @@ async def process_summary_task( user_id: int, chat_id: int, message_id_to_edit:
                 if final_summary.startswith("Error:") or final_summary.startswith("Sorry,"): user_feedback_message = final_summary; logger.warning(f"[Task {task_id}] Summary generation failed: {final_summary}")
                 else:
                     max_length = 4096; summary_parts = [final_summary[i:i+max_length] for i in range(0, len(final_summary), max_length)]
-                    await retry_bot_operation( bot.send_message, chat_id=chat_id, text=summary_parts[0], parse_mode=ParseMode.MARKDOWN, link_preview_options={'is_disabled': True} )
-                    for part in summary_parts[1:]: await asyncio.sleep(0.5); await retry_bot_operation( bot.send_message, chat_id=chat_id, text=part, parse_mode=ParseMode.MARKDOWN, link_preview_options={'is_disabled': True} )
+                    # Sending potentially unescaped summary directly - assuming AI follows instructions
+                    # ParseMode.MARKDOWN might interpret unintended things, test needed. Consider ParseMode=None if issues arise.
+                    await retry_bot_operation( bot.send_message, chat_id=chat_id, text=summary_parts[0], parse_mode=None, link_preview_options={'is_disabled': True} )
+                    for part in summary_parts[1:]: await asyncio.sleep(0.5); await retry_bot_operation( bot.send_message, chat_id=chat_id, text=part, parse_mode=None, link_preview_options={'is_disabled': True} )
                     success = True; logger.info(f"[Task {task_id}] Successfully sent summary ({len(summary_parts)} parts)."); user_feedback_message = None
             elif user_feedback_message: logger.warning(f"[Task {task_id}] Sending failure feedback: {user_feedback_message}"); await retry_bot_operation( bot.send_message, chat_id=chat_id, text=user_feedback_message, link_preview_options={'is_disabled': True} )
         except Exception as e: