Update app.py

app.py

@@ -5,7 +5,8 @@ import gradio as gr
 from datetime import datetime
 import agent
 from models import models
-import urllib.request
+import requests
+import io
 import uuid
 base_url="https://johann22-chat-diffusion.hf.space/" 
 
@@ -54,30 +55,39 @@ def run_gpt(in_prompt,history):
 
 
 def run(purpose,history,model_drop):
+    
+    #print(purpose)
+    #print(hist)
+    task=None
+    directory="./"
     if history:
         history=str(history).strip("[]")
     if not history:
         history = ""
-    try:
-        out_prompt = run_gpt(purpose,history)
-        out_prompt.strip()
-        print (out_prompt)
-    except Exception as e:
-        out_prompt = f"An Error Occured generating the prompt \n {e}"
+
+    #action_name, action_input = parse_action(line)
+    out_prompt = run_gpt(
+        purpose,
+        history,
+        
+        )
+
     yield ("",[(purpose,out_prompt)],None)
-    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)
+    #out_img = infer(out_prompt)
+    model=loaded_model[int(model_drop)]
+    out_img=model(out_prompt)
+    print(out_img)
+    url=f'https://johann22-chat-diffusion.hf.space/file={out_img}'
+    print(url)
+    uid = uuid.uuid4()
+    #urllib.request.urlretrieve(image, 'tmp.png')
+    #out=Image.open('tmp.png')
+    r = requests.get(url, stream=True)
+    if r.status_code == 200:
+        out = Image.open(io.BytesIO(r.content))
+    yield ("",[(purpose,out_prompt)],out)
         #return ("", [(purpose,history)])
-        return ("An Error Occured generating the image",[(purpose,out_prompt)],None)
+