Update app.py

app.py

@@ -115,10 +115,67 @@ def summarize(transcript_text, word_count, model_sel, lang_sel):
     
     return response.text
 
-def punctuate(transcript, model_sel):
+def punctuate(transcript_text, model_sel):
     client = genai.Client(api_key=GEMINI_API_KEY)
-    prompt = f"Restore puncutations of the transcript from an ASR model given. Maintain the original content. Only reply with the output. text: \n{transcript}"
+    prompt = f"""
+    Task: Punctuate the given Persian transcript text from an ASR model accurately according to Persian punctuation rules. Do not change any characters, correct mistakes, or modify the text in any way. You are ONLY allowed to add appropriate punctuations.
 
+    Guidelines:
+    1. Period (نقطه .)
+    Ends a sentence.
+
+    Used after abbreviations (e.g., آی.ب.ام.).
+
+    2. Comma (ویرگول ,)
+    Separates words/phrases in a list.
+
+    Used in compound or conditional sentences.
+
+    Prevents ambiguity (e.g., شلنگ مخصوص، آتش‌نشانی).
+
+    Surrounds parenthetical phrases (e.g., استیو جابز، بنیان‌گذار اپل،…).
+
+    Distinguishes adverbs (e.g., بعد از چندین ماه، ورزش کردم).
+
+    Separates repeated words (e.g., آن کشور، کشور خوبی است).
+
+    3. Semicolon (نقطه‌ویرگول ؛)
+    Connects related sentences when a full stop is too strong.
+
+    Separates clauses in complex lists (e.g., آلمان، ایتالیا و ژاپن؛ انگلیس، شوروی و آمریکا).
+
+    Used before explanatory phrases (e.g., فتوسنتز مهم است؛ یعنی…).
+
+    4. Colon (دونقطه :)
+    Introduces explanations, lists, or direct quotes (e.g., او گفت: «من آماده‌ام.»).
+
+    5. Ellipsis (سه‌نقطه …)
+    Indicates omitted words (e.g., فرهنگی، سیاسی، اجتماعی و …).
+
+    6. Parentheses (پرانتز ())
+    Encloses extra information, dates, or clarifications (e.g., جنگ جهانی دوم (۱۹۴۵)).
+
+    7. Quotation Marks (گیومه «»)
+    Encloses direct speech (e.g., او گفت: «سلام!»).
+
+    Highlights specific words (e.g., او را «نابغه» خواندند).
+
+    8. Question Mark (علامت سؤال ؟)
+    Ends direct questions (e.g., آیا آمدی؟).
+
+    9. Exclamation Mark (علامت تعجب !)
+    Expresses surprise, emphasis, or commands (e.g., چه پرنده زیبایی!).
+
+
+    Instructions:
+    Apply these punctuation rules without modifying the original text.
+
+    Do not correct typos, spelling, or grammar mistakes.
+
+    Only return the punctuated text as output—no explanations or additional comments.
+    Input:
+    {transcript_text}  
+    """   
     response = client.models.generate_content(
         model=model_sel,
         contents=[prompt]