Update app.py

app.py

@@ -21,6 +21,8 @@ import stability_sdk.interfaces.gooseai.generation.generation_pb2 as generation
 from datetime import datetime
 from google.oauth2 import service_account
 from googleapiclient.discovery import build
+import wget
+import urllib.request
 
 stability_api = client.StabilityInference(
     key=st.secrets["STABILITY_KEY"], #os.environ("STABILITY_KEY"), # key=os.environ['STABILITY_KEY'], # API Key reference.
@@ -82,7 +84,7 @@ openai.api_key = st.secrets["OPENAI_KEY"] #os.environ("OPENAI_KEY") #os.environ[
 date_time = str(datetime.now())
 
 def g_sheet_log(myinput, output):
-  SERVICE_ACCOUNT_FILE = '/content/gsearch-376307-f16abe212f26.json'
+  SERVICE_ACCOUNT_FILE = 'gsheet.json'
 
   credentials = service_account.Credentials.from_service_account_file(
       filename=SERVICE_ACCOUNT_FILE
@@ -175,6 +177,19 @@ if Input_type == 'TEXT':
 
         if ("gen_draw" in string_temp):
             try:
+                try:
+                    wget.download(openai_response(prompt))
+                    img2 = Image.open(wget.download(openai_response(prompt)))
+                    img2.show()
+                    rx = 'Image returned'
+                    g_sheet_log(mytext, rx)
+                except:
+                    urllib.request.urlretrieve(openai_response(prompt),"img_ret.png")
+                    img = Image.open("img_ret.png")
+                    img.show()
+                    rx = 'Image returned'
+                    g_sheet_log(mytext, rx)
+            except:
                 # Set up our initial generation parameters.
                 answers = stability_api.generate(
                 prompt = mytext,
@@ -208,25 +223,25 @@ if Input_type == 'TEXT':
                             rx = 'Image returned'
                             g_sheet_log(mytext, rx)
                             
-            except:        
-                st.write('image is being generated please wait...')
-                def extract_image_description(input_string):
-                    return input_string.split('gen_draw("')[1].split('")')[0]
-                prompt=extract_image_description(string_temp)
-                # model_id = "CompVis/stable-diffusion-v1-4"
-                model_id='runwayml/stable-diffusion-v1-5'
-                device = "cuda"
+            # except:        
+            #     st.write('image is being generated please wait...')
+            #     def extract_image_description(input_string):
+            #         return input_string.split('gen_draw("')[1].split('")')[0]
+            #     prompt=extract_image_description(string_temp)
+            #     # model_id = "CompVis/stable-diffusion-v1-4"
+            #     model_id='runwayml/stable-diffusion-v1-5'
+            #     device = "cuda"
 
 
-                pipe = StableDiffusionPipeline.from_pretrained(model_id, torch_dtype=torch.float16)
-                pipe = pipe.to(device)
+            #     pipe = StableDiffusionPipeline.from_pretrained(model_id, torch_dtype=torch.float16)
+            #     pipe = pipe.to(device)
 
-                # prompt = "a photo of an astronaut riding a horse on mars"
-                image = pipe(prompt).images[0]  
+            #     # prompt = "a photo of an astronaut riding a horse on mars"
+            #     image = pipe(prompt).images[0]  
                 
-                image.save("astronaut_rides_horse.png")
-                st.image(image)
-                # image
+            #     image.save("astronaut_rides_horse.png")
+            #     st.image(image)
+            #     # image
 
         elif ("vid_tube" in string_temp):
             s = Search(mytext)