Update app.py

app.py

@@ -718,59 +718,60 @@ elif Usage == 'Random Questions':
                 result = transcription["text"]
                 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 -''',
-                    temperature=0.49,
-                    max_tokens=256,
-                    top_p=1,
-                    frequency_penalty=0,
-                    presence_penalty=0
-                    )
-                    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 
-                    
-                    elif ("vid_tube" in string_temp):
-                        s = Search(question)
-                        search_res = s.results
-                        first_vid = search_res[0]
-                        print(first_vid)
-                        string = str(first_vid)
-                        video_id = string[string.index('=') + 1:-1]
-                        # print(video_id)
-                        YoutubeURL = "https://www.youtube.com/watch?v="
-                        OurURL = YoutubeURL + video_id
-                        st.write(OurURL)
-                        st_player(OurURL)
-        
-                    elif ("don't" in string_temp or "internet" in string_temp  ):
-                        st.write('*searching internet*')
-                        search_internet(question)
-                    else:
-                        st.write(string_temp)
+                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 -''',
+                temperature=0.49,
+                max_tokens=256,
+                top_p=1,
+                frequency_penalty=0,
+                presence_penalty=0
+                )
+                
+                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 
+                
+                elif ("vid_tube" in string_temp):
+                    s = Search(question)
+                    search_res = s.results
+                    first_vid = search_res[0]
+                    print(first_vid)
+                    string = str(first_vid)
+                    video_id = string[string.index('=') + 1:-1]
+                    # print(video_id)
+                    YoutubeURL = "https://www.youtube.com/watch?v="
+                    OurURL = YoutubeURL + video_id
+                    st.write(OurURL)
+                    st_player(OurURL)
+    
+                elif ("don't" in string_temp or "internet" in string_temp  ):
+                    st.write('*searching internet*')
+                    search_internet(question)
+                else:
+                    st.write(string_temp)
             
         # except:
         #     pass