Upload intro.py

page/intro.py

@@ -0,0 +1,139 @@
+import collections
+from numpy.core.defchararray import lower
+import streamlit as st
+import numpy as np
+import pandas as pd
+import zipfile
+import io
+import os
+from streamlit.elements.image import image_to_url
+import gzip
+import requests
+from io import BytesIO
+from PIL import Image, ImageDraw
+import base64
+import datetime
+import random, os, time
+import threading
+
+
+#List of files to use for each image object, 40 images total
+files = [None]*40 
+
+
+
+def randomFile(ix):
+
+    path = r"exampleImages"
+
+    dd = list(os.listdir(path))
+    random.shuffle(dd)
+
+    #Parse through each file in directory
+    for file in dd:
+
+        #If file is not in files list, use it for next image
+        if file not in files:
+
+            files[ix] = file
+            return "exampleImages/"+file
+       
+def gen(_p):
+
+
+    if(_p is not False):
+        st.session_state.prompt = _p
+        st.session_state.page = 0
+        return
+
+    _1 = ["A modern ","A post-modern ","A classical ", "A parametric ", "A contemporary ", "A minimalist "]
+    _2 = ["museum architecture","home architecture","landscape architecture","interior design","structural architecture"]
+    _3 = [""," in the style of I.M. Pei"," in the style of Frank Gehry"," in the style of John Lautner"," in the style of Frank Lloyd Wright"]
+    _4 = [" photograph",", watercolor painting",", oil painting", ", digital art"]
+
+    st.session_state.prompt = str(random.choice(_1)+random.choice(_2)+random.choice(_3)+random.choice(_4))
+    st.session_state.page = 0
+    
+
+def app():
+
+    #Array of image objects
+    images = []
+
+    for i in range(30):
+        files.append( randomFile(i) ) 
+
+    placeholder = st.empty()
+
+    with placeholder.container():
+
+        columns = [col1, col2, col3, col4, col5] = st.columns(5)
+
+        ix = 0
+        for column in columns:
+            with column:
+                for i in range(2):
+                    images.append( st.empty() )
+
+                    with images[ix].container():
+                        st.image("exampleImages/"+files[ix])
+
+                    ix += 1
+
+        
+        st.title('AI-Generated Architecture')
+   
+        prompt = st.text_input(label="Describe the architecture you want to see",value="An oil painting of a modern architecture")
+
+        c1,c2,c3 = st.columns(3)
+
+        with c1:
+            if st.button("Generate Architecture"):
+                if prompt:  
+                    gen(prompt)
+                elif prompt == "":
+                    gen(False)
+                return
+                
+    
+        with c2:
+            if st.button("Random Prompt"):
+                gen(False)
+                return
+
+        st.text("")
+        
+
+        columns2 = [col1, col2, col3, col4, col5] = st.columns(5)
+
+   
+        for column in columns2:
+            with column:
+                for i in range(4):
+
+                    images.append( st.empty() )
+
+                    with images[ix].container():
+                        st.image("exampleImages/"+files[ix])
+
+                    ix += 1
+
+        last = -1
+
+        while(True):
+            ch = random.randrange(30)
+            with images[ch].container():
+                st.image(randomFile(ch))
+            time.sleep(0.33)
+
+
+
+    #download_thread = threading.Thread(target=background, name="Downloader")
+    #download_thread.start()
+
+
+        
+
+
+    
+