Update app.py

app.py

@@ -17,11 +17,26 @@ VOICE_OPTIONS = [
 def inference(
     text,
     voice,
+    emotion_option,
     preset_option,
 ):
-
+    
     texts = [text]
 
+    Angry_tone = "[I am so angry]"
+    Sad_tone = "[I am so sad]"
+    Happy_tone = "[I am so happy]"
+    Scared_tone = "[I am so scared]"
+
+    if emotion_option == "Angry":
+        text = Angry_tone + text
+    if emotion_option == "Sad":
+        text = Sad_tone + text
+    if emotion_option == "Happy":
+        text = Happy_tone + text
+    if emotion_option == "Scared":
+        text = Scared_tone + text
+
     voices = [voice]
 
     if len(voices) == 1:
@@ -53,6 +68,11 @@ def main():
         VOICE_OPTIONS, value="jane_eyre", label="Select voice:", type="value"
     )
 
+    emotion_option = gr.Radio(
+        ["Angry", "Sad", "Happy", "Scared"],
+        type="value",
+    )
+
     preset_option = gr.Radio(
         ["ultra_fast", "fast", "standard", "high_quality"],
         label="ultra_fast for quick inference and high_quality for better inference",
@@ -66,6 +86,7 @@ def main():
         inputs=[
             text,
             voice,
+            emotion_option,
             preset_option,
         ],
         title=title,