Upload Visualization.py

pages/Visualization.py

@@ -0,0 +1,41 @@
+import streamlit as st
+import requests
+import io
+from PIL import Image
+import os
+
+st.set_page_config(page_title="Student Assistant")
+
+# API для генерации изображения
+API_URL_img = "https://api-inference.huggingface.co/models/CompVis/stable-diffusion-v1-4"
+headers = {"Authorization": os.getenv("api_token")}
+
+# Функция генерации изображения
+def generate_img(payload):
+	response = requests.post(API_URL_img, headers=headers, json=payload)
+	return response.content
+
+st.markdown('# :female-student: Персональный помощник для студентов')
+
+st.markdown("# :sparkles: Изучение английского языка через визуальное восприятие")
+
+image_idea = st.text_input('Предложите свою тему для генерации изображения', value="Astronaut riding a horse")
+image_gen__btn = st.button('Генерировать изображение')
+if image_gen__btn:
+    with st.spinner('...'):
+        image_bytes = generate_img(image_idea) 
+        image_raw = io.BytesIO(image_bytes)        
+        st.success('Готово')        
+    if image_raw is not None:        
+        st.image(image_raw, width=600)     
+        st.markdown('## План ответа поможет вам:') 
+        st.markdown('+ the place;') 
+        st.markdown('+ the action;') 
+        st.markdown('+ the person’s appearance;') 
+        st.markdown('+ whether you like the picture or not;') 
+        st.markdown('+ why.') 
+        st.markdown('Start with: “I’d like to describe this picture. The picture shows …” ')
+
+
+
+