Update app.py

app.py

@@ -1,14 +1,37 @@
 import streamlit as st
 from freeGPT import Client
+import requests
+import io
+from PIL import Image
+import os
+
+api_token = os.environ.get("api_token")
+API_URL = "https://api-inference.huggingface.co/models/goofyai/3d_render_style_xl"
+headers = {"Authorization": f"Bearer {api_token}"}
 
 st.title("Freak AI - Your stupid assistant")
 st.header("Type any message and get stupid response", divider="rainbow")
 messages = st.container(height=400)
 prompt = st.chat_input(placeholder="Eg. How are you today?")
 
+def query(payload):
+	response = requests.post(API_URL, headers=headers, json=payload)
+	return response.content
+
 if prompt:
-    resp = Client.create_completion("gpt3", prompt + " . Give answer as very stupid dude guy, as stupid and funny as it can be. I should laugh when i'll read answer")
-    user_msg = messages.chat_message("user", avatar="https://cdn.discordapp.com/attachments/1159192117751578705/1207020816332619796/kvikontent_professional_shot_of_clever_man_witha_book_in_hands__73d404c7-c515-4d49-b989-ac84dfe1e4ea.png?ex=65de20bb&is=65cbabbb&hm=1086ecaa2cfd107aa2912981226548f17bef10d54459f69d22176d38bdb1a05a&")
-    user_msg.write(prompt)
-    resp_msg = messages.chat_message("assistant", avatar="https://cdn.discordapp.com/attachments/1159192117751578705/1207019218466373642/kvikontent_professioanl_shot_of_stupid_man_c81e05bf-4530-497e-a9f3-63acdeea5345.png?ex=65de1f3f&is=65cbaa3f&hm=79516030ac8aa4d53f19c82a05cacd32ec0dceeb9b75d329a253bf4f3b8727a9&")
-    resp_msg.write(resp)
+    if "Gerenate" in promt or "Imagine" in prompt or "gemerate" in prompt or "imagine" in prompt or "Create" in prompt or "create" in prompt or "Image" in prompt:
+        image_bytes = query({
+    	    "inputs": prompt,
+        })
+        image = Image.open(io.BytesIO(image_bytes))
+
+        user_msg = messages.chat_message("user", avatar="https://cdn.discordapp.com/attachments/1159192117751578705/1207020816332619796/kvikontent_professional_shot_of_clever_man_witha_book_in_hands__73d404c7-c515-4d49-b989-ac84dfe1e4ea.png?ex=65de20bb&is=65cbabbb&hm=1086ecaa2cfd107aa2912981226548f17bef10d54459f69d22176d38bdb1a05a&")
+        user_msg.write(prompt)
+        resp_msg = messages.chat_message("assistant", avatar="https://cdn.discordapp.com/attachments/1159192117751578705/1207019218466373642/kvikontent_professioanl_shot_of_stupid_man_c81e05bf-4530-497e-a9f3-63acdeea5345.png?ex=65de1f3f&is=65cbaa3f&hm=79516030ac8aa4d53f19c82a05cacd32ec0dceeb9b75d329a253bf4f3b8727a9&")
+        resp_msg.image(image, caption=prompt)
+    else:
+        resp = Client.create_completion("gpt3", prompt + " . Give answer as very stupid dude guy, as stupid and funny as it can be. I should laugh when i'll read answer")
+        user_msg = messages.chat_message("user", avatar="https://cdn.discordapp.com/attachments/1159192117751578705/1207020816332619796/kvikontent_professional_shot_of_clever_man_witha_book_in_hands__73d404c7-c515-4d49-b989-ac84dfe1e4ea.png?ex=65de20bb&is=65cbabbb&hm=1086ecaa2cfd107aa2912981226548f17bef10d54459f69d22176d38bdb1a05a&")
+        user_msg.write(prompt)
+        resp_msg = messages.chat_message("assistant", avatar="https://cdn.discordapp.com/attachments/1159192117751578705/1207019218466373642/kvikontent_professioanl_shot_of_stupid_man_c81e05bf-4530-497e-a9f3-63acdeea5345.png?ex=65de1f3f&is=65cbaa3f&hm=79516030ac8aa4d53f19c82a05cacd32ec0dceeb9b75d329a253bf4f3b8727a9&")
+        resp_msg.write(resp)