Update app.py

app.py

@@ -8,6 +8,142 @@ ai_role_dict = {
 "sound_engineer": "You are an Experienced Sound Engineer, who can provide expert feedback on the arrangement being used."
 }
 
+languages = [
+"Afrikaans",
+"Albanian",
+"Amharic",
+"Arabic",
+"Armenian",
+"Assamese",
+"Aymara",
+"Azerbaijani",
+"Bhojpuri",
+"Basque",
+"Belarusian",
+"Bengali",
+"Bambara",
+"Bosnian",
+"Bulgarian",
+"Burmese (Myanmar)",
+"Catalan",
+"Cebuano",
+"Chewa (Chichewa)",
+"Chinese (Simplified)",
+"Chinese (Traditional)",
+"Corsican",
+"Croatian",
+"Czech",
+"Danish",
+"Dogri",
+"Dutch",
+"English",
+"Esperanto",
+"Estonian",
+"Ewe",
+"Finnish",
+"French",
+"Galician",
+"Georgian",
+"German",
+"Greek",
+"Guarani",
+"Gujarati",
+"Haitian Creole",
+"Hausa",
+"Hawaiian",
+"Hebrew",
+"Hindi",
+"Hmong",
+"Hungarian",
+"Icelandic",
+"Igbo",
+"Ilocano",
+"Indonesian",
+"Irish",
+"Italian",
+"Japanese",
+"Javanese",
+"Kannada",
+"Kazakh",
+"Khmer",
+"Kinyarwanda",
+"Konkani",
+"Korean",
+"Krio",
+"Kurdish (Kurmanji)",
+"Kurdish (Sorani)",
+"Kyrgyz",
+"Lao",
+"Latin",
+"Latvian",
+"Lingala",
+"Lithuanian",
+"Luganda",
+"Luxembourgish",
+"Macedonian",
+"Maithili",
+"Malagasy",
+"Malay",
+"Malayalam",
+"Maldivian (Dhivehi)",
+"Maltese",
+"Māori (Maori)",
+"Marathi",
+"Meitei (Manipuri, Meiteilon)",
+"Mizo",
+"Mongolian",
+"Nepali",
+"Northern Sotho (Sepedi)",
+"Norwegian (Bokmål)",
+"Odia (Oriya)",
+"Oromo",
+"Pashto",
+"Persian",
+"Polish",
+"Portuguese",
+"Punjabi (Gurmukhi)",
+"Quechua",
+"Romanian",
+"Russian",
+"Samoan",
+"Sanskrit",
+"Scottish Gaelic (Scots Gaelic)",
+"Serbian",
+"Shona",
+"Sindhi",
+"Sinhala",
+"Slovak",
+"Slovenian",
+"Somali",
+"Sotho (Sesotho)",
+"Spanish",
+"Sundanese",
+"Swahili",
+"Swedish",
+"Tagalog (Filipino)",
+"Tajik",
+"Tamil",
+"Tatar",
+"Telugu",
+"Thai",
+"Tigrinya",
+"Tsonga",
+"Turkish",
+"Turkmen",
+"Twi",
+"Ukrainian",
+"Urdu",
+"Uyghur",
+"Uzbek",
+"Vietnamese",
+"Welsh",
+"West Frisian (Frisian)",
+"Xhosa",
+"Yiddish",
+"Yoruba",
+"Zulu"
+]
+
 from tenacity import (
     retry,
     stop_after_attempt,
@@ -35,9 +171,9 @@ def write_intermediate_outputs(filename, text):
 
     return sample_file_path
 
-def write_and_compose(model, api_key, genre, keywords, emotion):
+def write_and_compose(model, api_key, language, genre, keywords, emotion):
     openai.api_key = api_key
-    initial_lyrics = get_response(ai_role_dict['freelance_lyricist'], "Write structured lyrics of a {} song with the following keywords - {}, and use the following emotion - {}".format(genre, keywords, emotion), model)
+    initial_lyrics = get_response(ai_role_dict['freelance_lyricist'], "Write structured lyrics of a {} {} song with the following keywords - {}, and use the following emotion - {}".format(language, genre, keywords, emotion), model)
 
     query_feedback = '''The Freelance Lyricist submitted these lyrics:
 
@@ -68,15 +204,15 @@ Incorporate this feedback, and make suggested changes to the lyrics based on the
 
     # Workflow: Step 4
 
-    query_composer = '''Given the lyrics of the {} song on {} in the emotion - {} -
+    query_composer = '''Given the lyrics of the {} {} song on {} in the emotion - {} -
 
 {}
 
 write a suitable chord progression (for each line of the same lyrics), followed by the suitable arrangement required to sing and record the song (in bullet points)'''
 
-    composition_1 = get_response(ai_role_dict['music_composer'], query_composer.format(genre, keywords, emotion, final_lyrics), model)
+    composition_1 = get_response(ai_role_dict['music_composer'], query_composer.format(language, genre, keywords, emotion, final_lyrics), model)
 
-    query_sound_engineer = '''Given the lyrics of the {} song on {} in the emotion - {} -
+    query_sound_engineer = '''Given the lyrics of the {} {} song on {} in the emotion - {} -
 
 {}
 
@@ -87,9 +223,9 @@ with a Chord Progression and Arrangement (suggested by the Music Composer) -
 could you write improvements that could be made to the Arrangement (in bullet points)? If the current arrangement is upto the mark, write "No change in the arrangement required"
 '''
 
-    composition_2 = get_response(ai_role_dict['sound_engineer'], query_sound_engineer.format(genre, keywords, emotion, final_lyrics, composition_1), model)    
+    composition_2 = get_response(ai_role_dict['sound_engineer'], query_sound_engineer.format(language, genre, keywords, emotion, final_lyrics, composition_1), model)    
 
-    final_query = '''Given the lyrics of the {} song on {} in the emotion - {} -
+    final_query = '''Given the lyrics of the {} {} song on {} in the emotion - {} -
 
 {}
 
@@ -108,7 +244,7 @@ and further improvements on the Arrangement (suggested by the Sound Engineer)
 
 '''
 
-    final_response = get_response(ai_role_dict['music_director'], final_query.format(genre, keywords, emotion, final_lyrics, composition_1, composition_2), model)    
+    final_response = get_response(ai_role_dict['music_director'], final_query.format(language, genre, keywords, emotion, final_lyrics, composition_1, composition_2), model)    
 
     final_improvements = final_response.split('==========')[0]
 
@@ -132,7 +268,7 @@ description = '''<span style="font-family:Papyrus; font-size:1.5em;">
 
 # Objective -
 
-Given specific Genre, Keywords, and Emotion, make a Brand New Song without lifting a finger!
+Given specific Language, Genre, Keywords, and Emotion of your choice, make a Brand New Song without lifting a finger!
 
 1. Get lyrics of a new song
 2. Get a suitable chord progression
@@ -151,7 +287,7 @@ Given specific Genre, Keywords, and Emotion, make a Brand New Song without lifti
 
 # Workflow (Intermediate outputs/results are output as downloadable files) -
 
-1. Get Inputs from user (OpenAI API Endpoint, API Key, keywords, genre, emotion for the song). Check out [this link](https://platform.openai.com/account/api-keys) to get your API Key
+1. Get Inputs from user (OpenAI API Endpoint, API Key, language, keywords, genre, emotion for the song). Check out [this link](https://platform.openai.com/account/api-keys) to get your API Key
 2. Experienced Freelance Lyricist writes a lyrics draft (**see `step_2.txt`**) 
 3. Experienced Music Director and Experienced Lyricist provide feedback (**see `step_3A.txt` & `step_3B.txt` respectively**)
 4. Experienced Freelance Lyricist incorporates the feedback, **Lyrics is finalized here**
@@ -163,7 +299,7 @@ Given specific Genre, Keywords, and Emotion, make a Brand New Song without lifti
 
 demo = gr.Interface(title = 'Write and Compose brand new Songs using an Elite *AI Music Team*', description = description,
     fn=write_and_compose,
-    inputs=[gr.Radio(["gpt-3.5-turbo", "gpt-4"], value="gpt-3.5-turbo", label = "Choose the OpenAI API Endpoint"), "text", "text", gr.Textbox(label="Keywords (separated by comma)"), "text"], # model, api_key, genre, keywords, emotion
+    inputs=[gr.Radio(["gpt-3.5-turbo", "gpt-4"], value="gpt-3.5-turbo", label = "Choose the OpenAI API Endpoint"), gr.Textbox(label="API Key (Check out [this link](https://platform.openai.com/account/api-keys) to get your API Key)"), gr.Dropdown(choices=languages, value='English', label="Language of the lyrics"), gr.Textbox(label="Genre"), gr.Textbox(label="Keywords (separated by comma)"), gr.Textbox(label="Emotion")], # model, api_key, language, genre, keywords, emotion
     # outputs=[gr.Textbox(label="Lyrics after Step #2"), gr.Textbox(label="Feedback provided by Music Director in Step #3"), gr.Textbox(label="Feedback provided by Lyricist in Step #3"), gr.Textbox(label="Final Lyrics of the song after Step #4"), gr.Textbox(label="Chord Progression and Arrangement suggested by Music Composer in Step #5"), gr.Textbox(label="Arrangement improvements suggested by Sound Engineer in Step #6"), gr.Textbox(label="Chord and Arrangement improvements suggested by Music Director in Step #7"), gr.Textbox(label="Final Chord Progression, Arrangment, and Song Title")], # initial_lyrics, feedback1, feedback2, final_lyrics, composition_1, composition_2, final_improvements, final_chord_prog_and_composition
     outputs=[gr.Textbox(label="Final Lyrics of the song after Step #4"), gr.Textbox(label="Final Chord Progression, Arrangement, and Song Title"), gr.File(label='Intermediate Outputs')], # initial_lyrics, feedback1, feedback2, final_lyrics, composition_1, composition_2, final_improvements, final_chord_prog_and_composition
 )