mirror of
https://github.com/qwertyforce/scenery.git
synced 2025-05-05 06:59:41 +00:00
Update reverse_search.tsx
This commit is contained in:
parent
188cb2c2df
commit
6e2838d858
@ -1,7 +1,8 @@
|
||||
import React, { useState } from 'react';
|
||||
/* eslint-disable @typescript-eslint/no-explicit-any */
|
||||
import React, { useEffect, useState } from 'react';
|
||||
import Box from '@material-ui/core/Box';
|
||||
import AppBar from '../components/AppBar'
|
||||
import { DropzoneArea } from 'material-ui-dropzone';
|
||||
import { DropzoneAreaBase } from 'material-ui-dropzone';
|
||||
import Button from '@material-ui/core/Button';
|
||||
import config from '../config/config'
|
||||
import axios from "axios"
|
||||
@ -9,23 +10,34 @@ import { useRouter } from 'next/router'
|
||||
import Backdrop from '@material-ui/core/Backdrop';
|
||||
import CircularProgress from '@material-ui/core/CircularProgress';
|
||||
import { makeStyles } from '@material-ui/core/styles';
|
||||
import TextField from '@material-ui/core/TextField';
|
||||
|
||||
const useStyles = makeStyles(() => ({
|
||||
backdrop: {
|
||||
zIndex: 9999,
|
||||
color: '#fff',
|
||||
},
|
||||
url_div:{
|
||||
display:"flex",
|
||||
marginLeft:"10px",
|
||||
marginBottom:"10px"
|
||||
},
|
||||
url_text_field:{
|
||||
width:"300px",
|
||||
marginRight:"10px"
|
||||
}
|
||||
}));
|
||||
|
||||
export default function ReverseSearch() {
|
||||
const classes = useStyles();
|
||||
const router = useRouter()
|
||||
const [Files, setFiles] = useState([]);
|
||||
const [url, setUrl] = useState("");
|
||||
const [fileObjects, setFileObjects] = useState([]);
|
||||
const [open, setOpen] = useState(false);
|
||||
const send_image = (token: string,mode:string) => {
|
||||
setOpen(true)
|
||||
const formData = new FormData();
|
||||
formData.append("image", Files[0]);
|
||||
formData.append("image", fileObjects[0].file);
|
||||
formData.append("g-recaptcha-response", token);
|
||||
formData.append("mode", mode);
|
||||
axios(`${config.reverse_search_url}/reverse_search`, {
|
||||
@ -52,6 +64,26 @@ export default function ReverseSearch() {
|
||||
});
|
||||
})
|
||||
}
|
||||
const get_image_by_url= async ()=>{
|
||||
setOpen(true)
|
||||
try{
|
||||
const x =await axios.get(url,{responseType:"blob"})
|
||||
setOpen(false)
|
||||
const file = new File([x.data], "image.png", {type:"image/png"});
|
||||
const reader = new FileReader();
|
||||
reader.readAsDataURL(x.data);
|
||||
reader.onloadend = function () {
|
||||
const base64data = reader.result;
|
||||
|
||||
setFileObjects( ([{data:base64data,file:file}]) as any)
|
||||
return;
|
||||
}
|
||||
}catch(err){
|
||||
setOpen(false)
|
||||
alert("error")
|
||||
}
|
||||
}
|
||||
|
||||
return (
|
||||
<div>
|
||||
<AppBar />
|
||||
@ -59,17 +91,31 @@ export default function ReverseSearch() {
|
||||
<CircularProgress color="inherit" />
|
||||
</Backdrop>
|
||||
<Box my={4}>
|
||||
<DropzoneArea
|
||||
<div className={classes.url_div}>
|
||||
<TextField onChange={(e)=>setUrl(e.target.value)} value={url} className={classes.url_text_field} label="url" variant="outlined" size="small" />
|
||||
<Button onClick={get_image_by_url} size="small" variant="outlined">Fetch</Button>
|
||||
</div>
|
||||
<DropzoneAreaBase
|
||||
acceptedFiles={['image/png', 'image/jpg', 'image/jpeg']}
|
||||
dropzoneText={"Drag and drop an image here or click"}
|
||||
fileObjects={fileObjects}
|
||||
filesLimit={1}
|
||||
onAdd={(newFileObjs:any) => {
|
||||
console.log(newFileObjs)
|
||||
console.log('onAdd', newFileObjs);
|
||||
setFileObjects([].concat(newFileObjs[0]));
|
||||
}}
|
||||
onDelete={(_removedFileObj, removedFileObjIdx) => {
|
||||
const remainingFileObjs = fileObjects.filter((_fileObject, i) => {
|
||||
return i !== removedFileObjIdx;
|
||||
});
|
||||
setFileObjects(remainingFileObjs)
|
||||
}}
|
||||
maxFileSize={49000000}
|
||||
onChange={(files) => setFiles((files as never))}
|
||||
/>
|
||||
</Box>
|
||||
<Button onClick={() => { _send_image("1") }} variant="contained" color="primary" >Reverse Search (fast, less accurate)</Button>
|
||||
<div style={{marginTop:"10px"}}><Button onClick={() => { _send_image("2") }} variant="contained" color="primary" >Reverse Search (slow, more accurate)</Button></div>
|
||||
|
||||
</div>
|
||||
);
|
||||
}
|
Loading…
x
Reference in New Issue
Block a user