Update app.py

app.py

@@ -83,6 +83,7 @@ from io import BytesIO
 from PIL import Image
 import gradio as gr
 import numpy as np
+import replicate
 import traceback
 import requests
 import openai
@@ -91,9 +92,10 @@ import base64
 import os
 import urllib.request
 
-def function(Textbox, Textbox2, Textbox3):
+def function(Textbox, Textbox2, Textbox3, Dropdown):
     target = os.environ.get("target")
     target2 = os.environ.get("target2")
+    os.environ["REPLICATE_API_TOKEN"] = os.environ.get("tk")
     # openai.api_key = target2
     # openai.api_base = os.environ.get("base")
     content = os.environ.get("content")
@@ -152,41 +154,81 @@ def function(Textbox, Textbox2, Textbox3):
                 print('IO Error:', str(e))
             
         if Textbox3 == target:
-            Textbox2 = Textbox2.replace("[", "").replace("]", "").replace("'", "")
-            Textbox2 = Textbox2.split(",")
-            Textbox2_edited = [x.strip() for x in Textbox2]
-            Textbox2_edited = list(Textbox2_edited)
-            Textbox2_edited.append(Textbox)
-            data = {
-                "messages": [
-                    {"role": "system", "content": content}
-                ]
-            }
-            for i in Textbox2_edited:
-                data["messages"].append(
-                    {"role": "user", "content": i}
-                )
-            try:
-                sleep(0.8)
-                # chat = openai.ChatCompletion.create(
-                #     model="gpt-3.5-turbo", messages=messages
-                # )
-                response = requests.post(target2, json=data)
-                reply = response.content.decode("utf-8")
-                # reply = reply.replace(" ", "%20")
-                # image_array = download_image(f"https://api.placid.app/u/pydav4ibo?quote[text]={reply}")
-                data["messages"].append({"role": "assistant", "content": reply})
-                return reply
-            except Exception as e:
-                print(traceback.format_exc())
-                return "Please Wait!"
+            if Dropdown == "Option1": 
+                Textbox2 = Textbox2.replace("[", "").replace("]", "").replace("'", "")
+                Textbox2 = Textbox2.split(",")
+                Textbox2_edited = [x.strip() for x in Textbox2]
+                Textbox2_edited = list(Textbox2_edited)
+                Textbox2_edited.append(Textbox)
+                data = {
+                    "messages": [
+                        {"role": "system", "content": content}
+                    ]
+                }
+                for i in Textbox2_edited:
+                    data["messages"].append(
+                        {"role": "user", "content": i}
+                    )
+                try:
+                    sleep(0.8)
+                    # chat = openai.ChatCompletion.create(
+                    #     model="gpt-3.5-turbo", messages=messages
+                    # )
+                    response = requests.post(target2, json=data)
+                    reply = response.content.decode("utf-8")
+                    # reply = reply.replace(" ", "%20")
+                    # image_array = download_image(f"https://api.placid.app/u/pydav4ibo?quote[text]={reply}")
+                    data["messages"].append({"role": "assistant", "content": reply})
+                    return reply
+                except Exception as e:
+                    print(traceback.format_exc())
+                    return "Please Wait!"
+            elif Dropdown == "Option2": 
+                Textbox2 = Textbox2.replace("[", "").replace("]", "").replace("'", "")
+                Textbox2 = Textbox2.split(",")
+                Textbox2_edited = [x.strip() for x in Textbox2]
+                Textbox2_edited = list(Textbox2_edited)
+                Textbox2_edited.append(Textbox)
+                data = {
+                    "messages": [
+                        {"role": "system", "content": content}
+                    ]
+                }
+                for i in Textbox2_edited:
+                    data["messages"].append(
+                        {"role": "user", "content": i}
+                    )
+                try:
+                    sleep(0.8)
+                    # chat = openai.ChatCompletion.create(
+                    #     model="gpt-3.5-turbo", messages=messages
+                    # )
+                    response = requests.post(target2, json=data)
+                    reply = response.content.decode("utf-8")
+                    # reply = reply.replace(" ", "%20")
+                    # image_array = download_image(f"https://api.placid.app/u/pydav4ibo?quote[text]={reply}")
+                    data["messages"].append({"role": "assistant", "content": reply})
+                    output = replicate.run(
+                        "suno-ai/bark:b76242b40d67c76ab6742e987628a2a9ac019e11d56ab96c4e91ce03b79b2787",
+                        input={
+                            "prompt": reply,
+                            "history_prompt": "en_speaker_5"
+                        }
+                    )
+                    output = output["audio_out"]
+                    return output
+                except Exception as e:
+                    print(traceback.format_exc())
+                    return "Please Wait!"
         else:
             return "Failed"
 
 inputs = [
     gr.inputs.Textbox(label="Textbox", type="text"),
     gr.inputs.Textbox(label="Textbox2", type="text"),
-    gr.inputs.Textbox(label="Textbox3", type="password")
+    gr.inputs.Textbox(label="Textbox3", type="password"),
+    gr.inputs.Dropdown(["Option1", "Option2"], label="Dropdown")
+
 ]
 
 # outputs = gr.outputs.Image(type="numpy")