Update app.py

app.py

@@ -88,7 +88,8 @@ import requests
 import openai
 import random
 import base64
-import os 
+import os
+import urllib.request
 
 def function(Textbox, Textbox2, Textbox3):
     target = os.environ.get("target")
@@ -138,16 +139,17 @@ def function(Textbox, Textbox2, Textbox3):
         else:
             return data
     else:
-        def send_request(reply):
+        def download_image(url):
             try:
-                response = requests.get(f"https://api.placid.app/u/pydav4ibo?quote[text]={reply}")
-                print(f'Response HTTP Status Code: {response.status_code}')
-                response.raise_for_status()
-                image = Image.open(BytesIO(response.content))
-                image_array = np.array(image)
-                return image_array
-            except requests.exceptions.RequestException as e:
-                print('HTTP Request failed:', str(e))
+                with urllib.request.urlopen(url) as response:
+                    image_data = response.read()
+                    image = Image.open(BytesIO(image_data))
+                    image_array = np.array(image)
+                    return image_array
+            except urllib.error.URLError as e:
+                print('URL Error:', str(e))
+            except IOError as e:
+                print('IO Error:', str(e))
             
         if Textbox3 == target:
             Textbox2 = Textbox2.replace("[", "").replace("]", "").replace("'", "")
@@ -162,14 +164,14 @@ def function(Textbox, Textbox2, Textbox3):
                 messages.append(
                     {"role": "user", "content": i}
                 )
-            try: 
+            try:
                 sleep(0.8)
                 chat = openai.ChatCompletion.create(
                     model="gpt-3.5-turbo", messages=messages
                 )
                 print(messages)
                 reply = chat.choices[0].message.content
-                image_array = send_request(reply)
+                image_array = download_image(reply)
                 messages.append({"role": "assistant", "content": reply})
                 return image_array
             except Exception as e:
@@ -191,6 +193,7 @@ iface = gr.Interface(fn=function, inputs=inputs, outputs=outputs)
 iface.launch()
 
 
+
 # from time import sleep
 # import gradio as gr
 # import traceback