Update app.py

app.py

@@ -107,6 +107,22 @@ st.write("**For convenience, you can use chatgpt to copy text and translate/eval
 st.write("-" * 50)
 
 
+popular_topics = [
+    "Sports", "Entertainment", "Politics", "Travel", "Technology",
+    "Health", "Business", "Science", "Education", "Lifestyle",
+    "Culture", "Environment", "Finance", "Food", "Gaming",
+    "History", "Law", "Literature", "Music", "News",
+    "Philosophy", "Religion", "Society", "World"
+]
+
+async def assign_topic(generated_text, topic_list=popular_topics):
+    lower_generated_text = generated_text.lower()
+    for topic in topic_list:
+        if topic.lower() in lower_generated_text:
+            return topic.title()
+    return generated_text
+
+    
 async def generate_from_api(user_input, generation_config):
     urls = [
         "https://pauljeffrey--sabiyarn-fastapi-app.modal.run/predict",
@@ -148,6 +164,7 @@ sample_texts = {
     "who are you?": "who are you?",
     "Speak Yoruba": "Speak Yoruba",
     "Translate 'how are you?' to Yoruba": "how are you?",
+    "Translate to pidgin": "Spain won the 2024 europa football cup. it was a tough one because they had to play very strong opponents in the quarter-finals, semi-finals and finals.",
     "Translate 'Often, all Yoruba children...' to Yoruba": "Often, all Yoruba children take pride in speaking the Yoruba language.",
     "Classify the sentiment": "Anyi na-echefu oke ike.",
     "what is the topic of this text": "Africa Free Trade Zone: Kò sí ìdènà láti kó ọjà láti orílẹ̀èdè kan sí òmíràn",
@@ -159,6 +176,7 @@ sample_texts = {
 instruction_wrap = {
     # "Translate 'Often, all Yoruba children...' to Yoruba":"<translate> Often, all Yoruba children take pride in speaking the Yoruba language. <yor>",
     "Tell me a story in pidgin": "<prompt> Tell me a story in pidgin <response>:",
+    "Spain won the 2024 europa football cup. it was a tough one because they had to play very strong opponents in the quarter-finals, semi-finals and finals.": "<translate> Spain won the 2024 europa football cup. it was a tough one because they had to play very strong opponents in the quarter-finals, semi-finals and finals. <pcm>"
     "how are you?": "<translate> how are you? <yor>:",
     "Often, all Yoruba children take pride in speaking the Yoruba language.": "<translate> Often, all Yoruba children take pride in speaking the Yoruba language. <yor>",
     "who are you?": "<prompt> who are you? <response>:",
@@ -271,6 +289,8 @@ if st.button("Generate"):
                 generated_text  = generated_text[:15]
                 print("split", generated_text.split(" ")[0],  re.split(r"\.|\n|\*\*|\*", generated_text)[0], generated_text.split(" "))
                 generated_text = re.split(r"\.|\n|\*\*|\*", generated_text)[0]
+
+                generated_text = asyncio.run(assign_topic(generated_text))
                 
             elif task == "Translation":
                 n_sentences = len(user_input.split("."))