Update chain_app.py

chain_app.py

@@ -11,7 +11,7 @@ openai_api_key = os.environ.get('OPENAI_API_KEY')
 groq_api_key = os.environ.get('GROQ_API_KEY')
 
 hf_text_client = Client("Artin2009/text-generation", hf_token=hf_token)
-hf_image_client = Client('Artin2009/image-generation')
+# hf_image_client = Client('Artin2009/image-generation')
 openai_client = OpenAI(api_key=openai_api_key)
 groq_client = Groq(api_key=groq_api_key)
 
@@ -38,10 +38,10 @@ async def chat_profile():
             name="Dorna-AI",
             markdown_description="One of the open-sourced models that neural brain team fine-tuned",
         ),
-        cl.ChatProfile(
-            name='Image-Generation',
-            markdown_description='Our image generation model, has a performance like midjourney',
-        ),
+        # cl.ChatProfile(
+        #     name='Image-Generation',
+        #     markdown_description='Our image generation model, has a performance like midjourney',
+        # ),
         cl.ChatProfile(
             name="GPT-4",
             markdown_description="OpenAI's GPT-4 model",
@@ -147,12 +147,12 @@ async def on_chat_start():
             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 == 'Image-Generation':
-        image = cl.Image(path='cat.png', name="result", display="inline")
-        await cl.Message(
-            content="I can make high quality & resoloution images for you, This is an example of what i can do!",
-            elements=[image],
-            ).send()
+    # if chat_profile == 'Image-Generation':
+    #     image = cl.Image(path='cat.png', name="result", display="inline")
+    #     await cl.Message(
+    #         content="I can make high quality & resoloution images for you, This is an example of what i can do!",
+    #         elements=[image],
+    #         ).send()
     if chat_profile == 'GPT-4':
         await cl.ChatSettings(
             [
@@ -439,23 +439,23 @@ async def main(message: cl.Message):
             content=model_response
         ).send()
     
-    elif chat_profile == 'Image-Generation':
-        result = hf_image_client.predict(
-		prompt=message.content,
-		negative_prompt="",
-		seed=0,
-		randomize_seed=True,
-		width=512,
-		height=512,
-		guidance_scale=0,
-		num_inference_steps=2,
-		api_name="/infer"
-        )
-        image = cl.Image(path=result, name="result", display="inline")
-        await cl.Message(
-            content="This message has an image!",
-            elements=[image],
-        ).send()
+  #   elif chat_profile == 'Image-Generation':
+  #       result = hf_image_client.predict(
+		# prompt=message.content,
+		# negative_prompt="",
+		# seed=0,
+		# randomize_seed=True,
+		# width=512,
+		# height=512,
+		# guidance_scale=0,
+		# num_inference_steps=2,
+		# api_name="/infer"
+  #       )
+  #       image = cl.Image(path=result, name="result", display="inline")
+  #       await cl.Message(
+  #           content="This message has an image!",
+  #           elements=[image],
+  #       ).send()
     elif chat_profile == 'GPT-4':
         completion = openai_client.chat.completions.create(
             model="gpt-4",