|
import cv2 |
|
import easyocr |
|
import gradio as gr |
|
from main import predict_mcq |
|
|
|
reader = easyocr.Reader(['th','en']) |
|
def ocr_with_easy(image): |
|
|
|
|
|
cv2.imwrite('image.png',image) |
|
image=cv2.imread('image.png') |
|
bounds = reader.readtext(image,paragraph="False",detail = 0) |
|
bounds = ''.join(bounds) |
|
return bounds |
|
|
|
def put_in_single_list(data): |
|
result=[] |
|
final_result=[] |
|
for i in data: |
|
result.append(i.get("question_statement")) |
|
result.append(i.get("answer")) |
|
result.append(i.get("options")) |
|
final_result.append(result) |
|
return final_result |
|
|
|
def MCQGenerator(image): |
|
I_text=ocr_with_easy(image) |
|
text={ |
|
"input_text":I_text |
|
} |
|
Mcqs=predict_mcq(text) |
|
data=Mcqs.get('questions') |
|
print(data) |
|
if data is not None: |
|
|
|
statement="" |
|
answer="" |
|
options="" |
|
for mcq in data: |
|
statement+=mcq.get('question_statement')+',' |
|
answer+=mcq.get('answer')+',' |
|
options+=mcq.get('options')[0]+','+mcq.get('options')[1]+','+mcq.get('options')[2]+',' |
|
return statement,answer,options |
|
else: |
|
return "Null","Null","Null" |
|
|
|
iface=gr.Interface(fn=MCQGenerator,inputs='image',outputs=[gr.components.Textbox(label="Question"),gr.components.Textbox(label="Answer"),gr.components.Textbox(label="Options")], |
|
examples=[['demo.PNG']]) |
|
iface.launch(debug=True) |
