Update app.py

app.py

@@ -389,13 +389,15 @@ elif Input_type == 'SPEECH':
                 st.text(question)
                 response = openai.Completion.create(
                 model="text-davinci-003",
-                prompt=f'''Your knowledge cutoff is 2021-09, and it is not aware of any events after that time. if the  
-                          Answer to following questions is not  from your knowledge base or in case of queries like weather 
-                            updates / stock updates / current news Etc which requires you to have internet connection  then print i don't have access to internet to answer your question, 
-                            if  question is related to  image or  painting or drawing generation then print ipython type output   function gen_draw("detailed prompt of image to be generated")
-                            if the question is related to playing a song or video or music of a singer then print ipython type output  function vid_tube("relevent search query")
-                            \nQuestion-{question}
-                            \nAnswer -''',
+                prompt=f'''Your name is HyperBot and  knowledge cutoff date is 2021-09, and you are not aware of any events after that time. if the  
+                    Answer to following questions is not from your knowledge base or in case of queries like date, time, weather 
+                      updates / stock updates / current affairs / news or people which requires you to have internet connection  then print i don't have access to internet to answer your question, 
+                      if  question is related to  image or  painting or drawing generation then print ipython type output function gen_draw("detailed prompt of image to be generated")
+                      if the question is related to playing a song or video or music of a singer then print ipython type output  function vid_tube("relevent search query")
+                      if the question is related to operating home appliances then print ipython type output function home_app(" action(ON/Off),appliance(TV,Geaser,Fridge,Lights,fans,AC)") . 
+                      if question is realted to sending mail or sms then print ipython type output function messenger_app(" message of us ,messenger(email,sms)")
+                      \nQuestion-{question}
+                      \nAnswer -''',
                 temperature=0.49,
                 max_tokens=256,
                 top_p=1,
@@ -405,24 +407,56 @@ elif Input_type == 'SPEECH':
                 string_temp=response.choices[0].text
     
                 if ("gen_draw" in string_temp):
-                    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)
-    
-                    # 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 
-                
+                    try:
+                        try:
+                            wget.download(openai_response(prompt))
+                            img2 = Image.open(wget.download(openai_response(prompt)))
+                            img2.show()
+                            rx = 'Image returned'
+                            now = datetime.datetime.now()
+                            date_time = now.strftime("%Y-%m-%d %H:%M:%S")
+                            csv_logs(mytext, rx, date_time)
+                        except:
+                            urllib.request.urlretrieve(openai_response(prompt),"img_ret.png")
+                            img = Image.open("img_ret.png")
+                            img.show()
+                            rx = 'Image returned'
+                            now = datetime.datetime.now()
+                            date_time = now.strftime("%Y-%m-%d %H:%M:%S")
+                            csv_logs(mytext, rx, date_time)
+                    except:
+                        # Set up our initial generation parameters.
+                        answers = stability_api.generate(
+                        prompt = mytext,
+                        seed=992446758, # If a seed is provided, the resulting generated image will be deterministic.
+                                # What this means is that as long as all generation parameters remain the same, you can always recall the same image simply by generating it again.
+                                # Note: This isn't quite the case for Clip Guided generations, which we'll tackle in a future example notebook.
+                        steps=30, # Amount of inference steps performed on image generation. Defaults to 30.
+                        cfg_scale=8.0, # Influences how strongly your generation is guided to match your prompt.
+                            # Setting this value higher increases the strength in which it tries to match your prompt.
+                            # Defaults to 7.0 if not specified.
+                        width=512, # Generation width, defaults to 512 if not included.
+                        height=512, # Generation height, defaults to 512 if not included.
+                        samples=1, # Number of images to generate, defaults to 1 if not included.
+                        sampler=generation.SAMPLER_K_DPMPP_2M # Choose which sampler we want to denoise our generation with.
+                                                            # Defaults to k_dpmpp_2m if not specified. Clip Guidance only supports ancestral samplers.
+                                                            # (Available Samplers: ddim, plms, k_euler, k_euler_ancestral, k_heun, k_dpm_2, k_dpm_2_ancestral, k_dpmpp_2s_ancestral, k_lms, k_dpmpp_2m)
+                        )
+            
+                        for resp in answers:
+                            for artifact in resp.artifacts:
+                                if artifact.finish_reason == generation.FILTER:
+                                    warnings.warn(
+                                        "Your request activated the API's safety filters and could not be processed."
+                                        "Please modify the prompt and try again.")
+                                if artifact.type == generation.ARTIFACT_IMAGE:
+                                    img = Image.open(io.BytesIO(artifact.binary))
+                                    st.image(img)
+                                    img.save(str(artifact.seed)+ ".png") # Save our generated images with their seed number as the filename.
+                                    rx = 'Image returned'
+                                    # g_sheet_log(mytext, rx)
+                                    csv_logs(mytext, rx, date_time)
+                    
                 elif ("vid_tube" in string_temp):
                     s = Search(question)
                     search_res = s.results
@@ -463,13 +497,15 @@ elif Input_type == 'SPEECH':
             question = result
             response = openai.Completion.create(
             model="text-davinci-003",
-            prompt=f'''Your knowledge cutoff is 2021-09, and it is not aware of any events after that time. if the  
-                      Answer to following questions is not  from your knowledge base or in case of queries like weather 
-                        updates / stock updates / current news Etc which requires you to have internet connection  then print i don't have access to internet to answer your question, 
-                        if  question is related to  image or  painting or drawing generation then print ipython type output   function gen_draw("detailed prompt of image to be generated")
-                        if the question is related to playing a song or video or music of a singer then print ipython type output  function vid_tube("relevent search query")
-                        \nQuestion-{question}
-                        \nAnswer -''',
+            prompt=f'''Your name is HyperBot and  knowledge cutoff date is 2021-09, and you are not aware of any events after that time. if the  
+                    Answer to following questions is not from your knowledge base or in case of queries like date, time, weather 
+                      updates / stock updates / current affairs / news or people which requires you to have internet connection  then print i don't have access to internet to answer your question, 
+                      if  question is related to  image or  painting or drawing generation then print ipython type output function gen_draw("detailed prompt of image to be generated")
+                      if the question is related to playing a song or video or music of a singer then print ipython type output  function vid_tube("relevent search query")
+                      if the question is related to operating home appliances then print ipython type output function home_app(" action(ON/Off),appliance(TV,Geaser,Fridge,Lights,fans,AC)") . 
+                      if question is realted to sending mail or sms then print ipython type output function messenger_app(" message of us ,messenger(email,sms)")
+                      \nQuestion-{question}
+                      \nAnswer -''',
             temperature=0.49,
             max_tokens=256,
             top_p=1,
@@ -480,23 +516,56 @@ elif Input_type == 'SPEECH':
             string_temp=response.choices[0].text
 
             if ("gen_draw" in string_temp):
-                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)
-
-                # 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 
+                try:
+                    try:
+                        wget.download(openai_response(prompt))
+                        img2 = Image.open(wget.download(openai_response(prompt)))
+                        img2.show()
+                        rx = 'Image returned'
+                        now = datetime.datetime.now()
+                        date_time = now.strftime("%Y-%m-%d %H:%M:%S")
+                        csv_logs(mytext, rx, date_time)
+                    except:
+                        urllib.request.urlretrieve(openai_response(prompt),"img_ret.png")
+                        img = Image.open("img_ret.png")
+                        img.show()
+                        rx = 'Image returned'
+                        now = datetime.datetime.now()
+                        date_time = now.strftime("%Y-%m-%d %H:%M:%S")
+                        csv_logs(mytext, rx, date_time)
+                except:
+                    # Set up our initial generation parameters.
+                    answers = stability_api.generate(
+                    prompt = mytext,
+                    seed=992446758, # If a seed is provided, the resulting generated image will be deterministic.
+                            # What this means is that as long as all generation parameters remain the same, you can always recall the same image simply by generating it again.
+                            # Note: This isn't quite the case for Clip Guided generations, which we'll tackle in a future example notebook.
+                    steps=30, # Amount of inference steps performed on image generation. Defaults to 30.
+                    cfg_scale=8.0, # Influences how strongly your generation is guided to match your prompt.
+                        # Setting this value higher increases the strength in which it tries to match your prompt.
+                        # Defaults to 7.0 if not specified.
+                    width=512, # Generation width, defaults to 512 if not included.
+                    height=512, # Generation height, defaults to 512 if not included.
+                    samples=1, # Number of images to generate, defaults to 1 if not included.
+                    sampler=generation.SAMPLER_K_DPMPP_2M # Choose which sampler we want to denoise our generation with.
+                                                        # Defaults to k_dpmpp_2m if not specified. Clip Guidance only supports ancestral samplers.
+                                                        # (Available Samplers: ddim, plms, k_euler, k_euler_ancestral, k_heun, k_dpm_2, k_dpm_2_ancestral, k_dpmpp_2s_ancestral, k_lms, k_dpmpp_2m)
+                    )
+    
+                    for resp in answers:
+                        for artifact in resp.artifacts:
+                            if artifact.finish_reason == generation.FILTER:
+                                warnings.warn(
+                                    "Your request activated the API's safety filters and could not be processed."
+                                    "Please modify the prompt and try again.")
+                            if artifact.type == generation.ARTIFACT_IMAGE:
+                                img = Image.open(io.BytesIO(artifact.binary))
+                                st.image(img)
+                                img.save(str(artifact.seed)+ ".png") # Save our generated images with their seed number as the filename.
+                                rx = 'Image returned'
+                                # g_sheet_log(mytext, rx)
+                                csv_logs(mytext, rx, date_time)
+ 
             
             elif ("vid_tube" in string_temp):
                 s = Search(question)