Update chain_app.py

chain_app.py

@@ -49,6 +49,10 @@ async def chat_profile():
         #     name='Image-Generation',
         #     markdown_description='Our image generation model, has a performance like midjourney',
         # ),
+        cl.ChatProfile(
+            name="gpt4-o-mini",
+            markdown_description="The best state of the art openai model",
+        ),
         cl.ChatProfile(
             name="GPT-4",
             markdown_description="OpenAI's GPT-4 model",
@@ -162,6 +166,28 @@ async def on_chat_start():
         await cl.Message(
             content='my name is Dorna, Your AI Assistant designed by neural nexus team. i was made by Artin Daneshvar and Sadra Noadoust, 2 iranian students!'
         ).send()
+    if chat_profile == 'gpt4-o-mini':
+        await cl.ChatSettings(
+            [
+                Select(
+                    id="OpenAI-Model",
+                    label="OpenAI - Model",
+                    values=["gpt-4"],
+                    initial_index=0,
+                ),
+                Slider(
+                    id="Temperature",
+                    label="Model Temperature",
+                    initial=0.7,
+                    min=0,
+                    max=1,
+                    step=0.1,
+                ),
+            ]
+        ).send()
+        await cl.Message(
+            content="Im one of the best models openai have released and i am configured by two iranian boys to help you."
+        )
 
     # if chat_profile == 'Image-Generation':
     #     image = cl.Image(path='cat.png', name="result", display="inline")
@@ -453,7 +479,18 @@ async def main(message: cl.Message):
         await cl.Message(
             content=model_response
         ).send()
-    
+    elif chat_profile == "gpt4-o-omni":
+        completion = openai_client.chat.completions.create(
+            model="gpt-4o-mini",
+            messages=[
+                {"role": "system", "content": "You are neural nexus official chatbot, you are made by Artin Daneshvar and Sadra Noadoust"},
+                {"role": "user", "content": message.content}
+            ]
+        )
+        model_response = completion.choices[0].message.content
+        await cl.Message(
+            content=model_response
+        ).send()
   #   elif chat_profile == 'Image-Generation':
   #       result = hf_image_client.predict(
 		# prompt=message.content,