def post_url(request: Request, files: str = Form(...)):
print(type(files))
try:
resp = requests.get(files, stream=True)
if not "image" in resp.headers['content-type']:
del resp
raise HTTPException(status_code=404, detail="Not an image file")
a = urlparse(files)
name = basename(a.path)
with open(cwd + f"\\static\\userFiles\\{name}", 'wb') as local_file:
resp.raw.decode_content = True
copyfileobj(resp.raw, local_file)
print(f"name => {name.split('.')[0]} ")
segment(cwd + f"\\static\\userFiles\\{name}", f"{name.split('.')[0]}")
# if isfile(url):
# remove(url)
name = name.split('.')[0]
except:
return {'error': "Something went wrong while segmeting url"}
# return files
return templates.TemplateResponse(
"segment_result.html", {
"request": request,
"code": f"/static/codes.txt",
"image": f"/static/segment/{name}.png",
"file": f"/static/segment/{name}.pt"
})
def post_url(request: Request, files: UploadFile = File(...)):
print(type(files))
try:
if not "image" in str(files.content_type):
raise HTTPException(status_code=404, detail="Not an image file")
except:
raise HTTPException(status_code=404, detail="Not an image file")
with open(cwd + f"\\static\\userFiles\\{files.filename}", "wb") as buffer:
copyfileobj(files.file, buffer)
name = files.filename.split(".")[0]
segment(cwd + f"\\static\\userFiles\\{files.filename}", f'{name}')
# if isfile(url):
# remove(url)
# return files
return templates.TemplateResponse(
"segment_result.html", {
"request": request,
"code": f"/static/codes.txt",
"image": f"/static/segment/{name}.png",
"file": f"/static/segment/{name}.pt"
})
def post_url(request: Request, files: str = Form(...)):
print(type(files))
try:
resp = requests.get(files, stream=True)
if not "image" in resp.headers['content-type']:
del resp
raise HTTPException(status_code=404, detail="Not an image file")
a = urlparse(files)
name = basename(a.path)
with open(cwd + f"\\static\\userFiles\\{name}", 'wb') as local_file:
resp.raw.decode_content = True
copyfileobj(resp.raw, local_file)
pred, url = classify(cwd + f"\\static\\userFiles\\{name}")
# if isfile(url):
# remove(url)
except Exception as e:
return {"Something went wrong while classifying url": f'{e}'}
# return files
return templates.TemplateResponse(
"classify_result.html", {
"request": request,
"image": f"/static/userFiles/{name}",
"file": name,
"prediction": pred
})
def post_image(request: Request, files: UploadFile = File(...)):
try:
if not "image" in str(files.content_type):
raise HTTPException(status_code=404, detail="Not an image file")
except:
raise HTTPException(status_code=404, detail="Not an image file")
with open(cwd + f"\\static\\userFiles\\{files.filename}", "wb") as buffer:
copyfileobj(files.file, buffer)
pred, url = classify(cwd + f"\\static\\userFiles\\{files.filename}")
# if isfile(url):
# remove(url)
# print(f"pred => {pred}")
# print(f"url => {url}")
return templates.TemplateResponse(
"classify_result.html", {
"request": request,
"image": f"/static/userFiles/{files.filename}",
"file": files.filename,
"prediction": pred
})
def read_url(request: Request):
return templates.TemplateResponse("segment_url_form.html", {
"request": request,
"code": "/static/codes.txt"
})
def read_url(request: Request):
return templates.TemplateResponse("classify_url_form.html",
{"request": request})
def read_image(request: Request):
return templates.TemplateResponse("classification_form.html",
{"request": request})
def read_item_with_id(request: Request, id: int, q: Optional[str] = None):
return templates.TemplateResponse("item.html", {
"request": request,
"id": id,
'q': q
})
def read_item(request: Request, q: Optional[str] = None):
return templates.TemplateResponse("item.html", {
"request": request,
"id": "None",
'q': q
})
def home(request: Request):
content = r"go to /item/{id}?q={query+string}"
return templates.TemplateResponse("home.html", {
"request": request,
"content": content
})
