Update app.py

no more prompt in the summary

app.py

@@ -63,9 +63,9 @@ def summarize(file, text, style, length):
     # System prompt based on language and style
     prompt_map = {
         "en": {
-            "Precise": "In English, distill the following text into a concise summary, utilizing formal and academic language to convey the essential information:",
-            "Sloppy": "In English, provide a brief and informal summary of the following text, using straightforward language to facilitate easy comprehension:",
-            "Keywords": "In English, condense the following text into a list of keywords, highlighting key points and main ideas in a clear and objective manner:",
+            "Precise": "Summarize concisely in a dry, scientific and academic style:",
+            "Sloppy": "Summarize comprehensively for a child:",
+            "Keywords": "Write only a list of important keywords of main ideas separated by a comma:",
         }#, <-- don't forget the comma!!!!!
         #"foo": { "precise": "another language or prompt map could go here"}
     }
@@ -96,6 +96,12 @@ def summarize(file, text, style, length):
         no_repeat_ngram_size=3
     )
     summary = tokenizer.decode(summary_ids[0], skip_special_tokens=True)
+    # removing the prompt text from the summary
+    prompt_text = prompt_map.get(lang_code, prompt_map["en"])[style]
+    if summary.startswith(prompt_text):
+        summary = summary[len(prompt_text):].strip()
+    if text_input in summary:
+        summary = summary.replace(text_input, "").strip()
 
     # These lines calculate and store the word count of the original text, 
     # the word count of the summary, and the percentage reduction in length after summarization.
@@ -131,7 +137,7 @@ with gr.Blocks() as demo:
 
             with gr.Row(): # for inline horizontal layout
                 style = gr.Dropdown(["Precise", "Sloppy", "Keywords"], label="Style")
-                length = gr.Radio(["Short", "Middle", "Long"], label="Length")
+                length = gr.Dropdown(["Short", "Middle", "Long"], label="Length")
                 token_info = gr.Text(label="Max. Tokens:", value="1024 tokens ~ 750–800 words", interactive=False)
                 btn = gr.Button("Transform")
         
@@ -149,4 +155,4 @@ with gr.Blocks() as demo:
         outputs=[summary, keywords, original_len, summary_len, reduction, plot]
     )
 
-demo.launch()
+demo.launch(share=True)