Update app.py

app.py

@@ -57,16 +57,24 @@ def run(purpose,history,model_drop):
         history=str(history).strip("[]")
     if not history:
         history = ""
-    out_prompt = run_gpt(purpose,history)
+    try:
+        out_prompt = run_gpt(purpose,history)
+    except Exception as e:
+        out_prompt = f"An Error Occured generating the prompt \n {e}"
     yield ("",[(purpose,out_prompt)],None)
-    model=loaded_model[int(model_drop)]
-    out_img=model(out_prompt)
-    print(out_img)
-    image=f'{base_url}file={out_img}'
-    uid = uuid.uuid4()
-    urllib.request.urlretrieve(image, f'{uid}.png')
-    return ("",[(purpose,out_prompt)],f'{uid}.png')
+    try:
+        model=loaded_model[int(model_drop)]
+        out_img=model(out_prompt)
+        print(out_img)
+        image=f'{base_url}file={out_img}'
+        uid = uuid.uuid4()
+        urllib.request.urlretrieve(image, f'{uid}.png')
+        return ("",[(purpose,out_prompt)],f'{uid}.png')
+
+    except Exception as e:
+        print (e)
         #return ("", [(purpose,history)])
+        return ("An Error Occured generating the image",[(purpose,out_prompt)],None)