fix: remove unwanted quotes from translation output and simplify prompts

Author: soukron

app/summarizer.py

@@ -15,7 +15,7 @@ def setup_argparse():
         "-u", "--url", required=True, help="URL of the document to summarize"
     )
     parser.add_argument(
-        "-l", "--language", 
+        "-t", "--translate", 
         help="Target language for translation (if specified, translation is enabled)"
     )
     return parser.parse_args()
@@ -30,22 +30,11 @@ def load_document(url):
 def setup_summarization_chain():
     """Setup the summarization chain with a prompt template and ChatOllama."""
     prompt_template = PromptTemplate(
-        template="""As a professional summarizer, create a detailed and comprehensive summary of the provided text, be it an article, post, conversation, or passage, while adhering to these guidelines:
-            1. Craft a summary that is detailed, thorough, in-depth, and complex, while maintaining clarity.
+        template="""Create a detailed and comprehensive summary of the following text. Provide only the summary without any headers, quotes, or additional text:
 
-            2. Incorporate main ideas and essential information, eliminating extraneous language and focusing on critical aspects.
+{text}
 
-            3. Rely strictly on the provided text, without including external information.
-
-            4. Format the summary in paragraph form for easy understanding.
-
-            5.Conclude your notes with [End of Notes, Message #X] to indicate completion, where "X" represents the total number of messages that I have sent. In other words, include a message counter where you start with #1 and add 1 to the message counter every time I send a message.
-
-        By following this optimized prompt, you will generate an effective summary that encapsulates the essence of the given text in a clear, detailed, and reader-friendly manner. Optimize output as markdown file.
-
-        "{text}"
-
-        DETAILED SUMMARY:""",
+SUMMARY:""",
         input_variables=["text"],
     )
 
@@ -57,11 +46,11 @@ def setup_summarization_chain():
 def setup_translation_chain(target_language="Spanish"):
     """Setup the translation chain with a prompt template and ChatOllama."""
     prompt_template = PromptTemplate(
-        template="""As a professional translator, provide a detailed and comprehensive translation of the provided text into {target_language}, ensuring that the translation is accurate, coherent, and faithful to the original text.
+        template="""Translate the following text into {target_language}. Provide only the translation without any quotes, headers, or additional text. The output should be clean and direct:
 
-        "{text}"
+{text}
 
-        DETAILED TRANSLATION:""",
+TRANSLATION:""",
         input_variables=["text", "target_language"],
     )
 
@@ -92,12 +81,12 @@ def main():
     print()
 
     # Translate if language is specified
-    if args.language:
+    if args.translate:
         print("=" * 60)
-        print(f"TRANSLATION TO {args.language.upper()}")
+        print(f"TRANSLATION TO {args.translate.upper()}")
         print("=" * 60)
         try:
-            translation = translate_text(summary, args.language)
+            translation = translate_text(summary, args.translate)
             print(translation)
         except Exception as e:
             print(f"Translation failed: {e}")