Update app.py

app.py

@@ -32,6 +32,13 @@ from fastapi import Form
 class Query(BaseModel):
     text: str
     code:str
+    host:str
+    
+class Query2(BaseModel):
+    text: str
+    code:str
+    filename:str
+    host:str
    
 # device = torch.device('cuda' if torch.cuda.is_available() else 'cpu')
 # tokenizer = BertTokenizerFast.from_pretrained('mrm8488/bert-small2bert-small-finetuned-cnn_daily_mail-summarization')
@@ -88,6 +95,7 @@ async def get_answer(q: Query ):
 
     text = q.text
     code= q.code
+    host= q.host
     
     
     N = 20
@@ -100,7 +108,7 @@ async def get_answer(q: Query ):
 
     filename= res
 
-    t = threading.Thread(target=do_ML, args=(filename,text,code))  
+    t = threading.Thread(target=do_ML, args=(filename,text,code,host))  
     t.start()
 
     return JSONResponse({"id": filename})
@@ -108,6 +116,25 @@ async def get_answer(q: Query ):
     return "hello"
 
 
+
+
+@app.post("/error")
+async def get_answer(q: Query2 ):
+
+    text = q.text
+    code= q.code
+    filename= q.filename
+    host= q.host
+
+
+    t = threading.Thread(target=do_ML, args=(filename,text,code,host))  
+    t.start()
+
+    return JSONResponse({"id": filename})
+
+
+
+
 import requests
 import io
 import torch
@@ -120,33 +147,36 @@ client = InferenceClient()
 # client = InferenceClient(model="SG161222/Realistic_Vision_V1.4")
 
     
-def do_ML(filename:str,text:str,code:str):
+def do_ML(filename:str,text:str,code:str,host:str):
+    try:
+        global client
 
-    global client
+        imagei = client.text_to_image(text)
 
-    imagei = client.text_to_image(text)
-
-    byte_array = io.BytesIO()
-    imagei.save(byte_array, format='JPEG')
-    image_bytes = byte_array.getvalue()
+        byte_array = io.BytesIO()
+        imagei.save(byte_array, format='JPEG')
+        image_bytes = byte_array.getvalue()
 
     
-    files = {'file': image_bytes}
+        files = {'file': image_bytes}
 
-    global audio_space
-    url = audio_space+code
+        global audio_space
+        url = audio_space+code
 
-    data = {"filename": filename}
-    response = requests.post(url, files=files,data= data)
+        data = {"filename": filename}
+        response = requests.post(url, files=files,data= data)
 
-    print(response.text)
+        print(response.text)
 
-    if response.status_code == 200:
-        print("File uploaded successfully.")
+        if response.status_code == 200:
+            print("File uploaded successfully.")
     # Handle the response as needed
-    else:
-        print("File upload failed.")
-
+        else:
+            print("File upload failed.")
 
+    except:
+        data={"text":text,"filename":filename}
+        requests.post(host+"texttoimage2handleerror",data=data)
+