Create app.py

app.py

@@ -0,0 +1,100 @@
+import streamlit as st
+import os
+import io
+import PIL
+import freeGPT
+
+hf_token = os.environ.get("API_TOKEN")
+API_URL = "https://api-inference.huggingface.co/models/Lykon/dreamshaper-xl-v2-turbo"
+headers = {"Authorization": f"Bearer {hf_token}"}
+
+st.title("🌎 AI Browser")
+st.write("Browser that gives results using ai")
+prompt = st.input_text("Search something", placeholder="Enter search query")
+search_btn = st.button("Search")
+results = st.container("Results")
+texts, images = st.tabs(["All", "Images"])
+
+def query(payload):
+      response = requests.post(API_URL, headers=headers, json=payload)
+      return response.content
+        
+def generate_image(prompt):
+    image_bytes = query({
+        "inputs": prompt,
+    })
+    image = Image.open(io.BytesIO(image_bytes))
+    return image
+
+if search_btn:
+    prompt1 = prompt + ". Give short answer to this question"
+    prompt2 = prompt + ". Give short funny answer to this question"
+    prompt3 = prompt + ". Give short documental answer to this question"
+
+    result1 = Client.create_completion("gpt3", prompt1)
+    result2 = Client.create_completion("gpt3", prompt2)
+    result3 = Client.create_completion("gpt3", prompt3)
+    result4 = Client.create_completion("gpt3", prompt4)
+    result5 = Client.create_completion("gpt3", prompt5)
+    result6 = Client.create_completion("gpt3", prompt6)
+
+    imageresult1 = generate_image(prompt + ". Create image of this in realistic style")
+    imageresult2 = generate_image(prompt + ". Create image of this in anime style")
+    imageresult3 = generate_image(prompt + ". Create image of this in black and white style")
+    imageresult4 = generate_image(prompt + ". Create image of this in photo style")
+    imageresult5 = generate_image(prompt + ". Create image of this in style of iphone low quality photo")
+    imageresult6 = generate_image(prompt + ". Create image of this in cinemtic style")
+    imageresult7 = generate_image(prompt + ". Create image of this in unreal engine style")
+    imageresult8 = generate_image(prompt + ". Create image of this in google style")
+    imageresult9 = generate_image(prompt + ". Create image of this in russia style")
+    imageresult10 = generate_image(prompt + ". Create image of this in chinese style")
+    imageresult11 = generate_image(prompt + ". Create image of this in style of sumsung low quality photo")
+    imageresult12 = generate_image(prompt + ". Create image of this in romantic style")
+
+    with texts:
+        st.write("We found these results on your query: ")
+        st.header(result1.split()[0], divider='rainbow')
+        st.text(result1)
+        st.header(result2.split()[0], divider='rainbow')
+        st.text(result2)
+        st.header(result3.split()[0], divider='rainbow')
+        st.text(result3)
+        st.header(result4.split()[0], divider='rainbow')
+        st.text(result4)
+        st.header(result5.split()[0], divider='rainbow')
+        st.text(result5)
+        st.header(result6.split()[0], divider='rainbow')
+        st.text(result6)
+        st.caption("That's the end!")
+
+    with images:
+        im1, im2, im3 = st.column(3)
+        im4, im5, im6 = st.column(3)
+        im9, im8, im7 = st.column(3)
+        im10, im11, im12 = st.column(3)
+
+        st.write("We found some images on your query: ")
+        with im1:
+            st.image(imageresult1)
+        with im2:
+            st.image(imageresult2)
+        with im3:
+            st.image(imageresult3)
+        with im4:
+            st.image(imageresult4)
+        with im5:
+            st.image(imageresult5)
+        with im6:
+            st.image(imageresult6)
+        with im7:
+            st.image(imageresult7)
+        with im8:
+            st.image(imageresult8)
+        with im9:
+            st.image(imageresult9)
+        with im10:
+            st.image(imageresult10)
+        with im11:
+            st.image(imageresult11)
+        with im12:
+            st.image(imageresult12)