Mingyang

pet-gui.py

@@ -1,4 +1,4 @@
-from fastapi import FastAPI, File, UploadFile,Request,status
+from fastapi import FastAPI, File, Form, UploadFile,Request,status
 from transformers import T5Tokenizer, T5ForConditionalGeneration
 from io import StringIO
 from fastapi.responses import HTMLResponse
@@ -7,7 +7,10 @@
 from fastapi.responses import FileResponse
 import logging
 import tarfile
+import json
 from fastapi.responses import RedirectResponse
+from fastapi.encoders import jsonable_encoder
+
 
 app = FastAPI()
 
@@ -23,22 +26,59 @@ async def read_root():
 async def get_form(request: Request):
     return templates.TemplateResponse("index.html", {"request": request})
 
-@app.get("/racia", response_class=HTMLResponse,name = "racia")
+@app.get("/training", response_class=HTMLResponse,name = "training")
 async def read_item(request: Request):
     num = 100
-    return templates.TemplateResponse("progress.html", {"request": request, "num": num})
-
+    return templates.TemplateResponse("next.html", {"request": request, "num": num})
+
+# @app.post("/basic")
+# async def get_form(request: Request,sample: str = Form(...), label: str = Form(...),templates: str = Form(...),one: str = Form(...), two: str = Form(...),model_para: str = Form(...),file: UploadFile = File(...)):
+#     file_upload = tarfile.open(fileobj=file.file, mode="r:gz")
+#     file_upload.extractall('./data_uploaded')
+#     print(f'sample:{sample}')
+#     print(f'label:{label}')
+#     print(f'sample:{templates}')
+#     print(f'1:{one}')
+#     print(f'2:{two}')
+#     print(f'model_para:{model_para}')
+#     para_dic = {"sample":sample,"label":label,"templates":templates,"one":one,"two":two,"model_para":model_para}
+#     with open('data.json', 'w') as f:
+#         json.dump(para_dic, f)
+#     redirect_url = request.url_for('racia')
+#     return RedirectResponse(redirect_url, status_code=303)
 @app.post("/basic")
 async def get_form(request: Request,file: UploadFile = File(...)):
     file_upload = tarfile.open(fileobj=file.file, mode="r:gz")
     file_upload.extractall('./data_uploaded')
-    redirect_url = request.url_for('racia')
+    da = await request.form()
+    da = jsonable_encoder(da)
+    templates_counter = 1
+    origin_counter = 1
+    mapping_counter = 1
+    para_dic = {"sample": da["sample"], "label": da["label"], "templates": da["templates"], "origin": da["origin"], "mapping": da["mapping"],
+                "model_para": da["model_para"]}
+    while f"templates_{str(templates_counter)}" in da:
+        template_key = f"templates_{str(templates_counter)}"
+        para_dic[template_key] = da[template_key]
+        templates_counter = templates_counter+1
+    while f"origin_{str(origin_counter)}" in da:
+        origin_key = f"origin_{str(origin_counter)}"
+        para_dic[origin_key] = da[origin_key]
+        origin_counter = origin_counter+1
+    while f"mapping_{str(mapping_counter)}" in da:
+        mapping_key = f"mapping_{str(mapping_counter)}"
+        para_dic[mapping_key] = da[mapping_key]
+        mapping_counter = mapping_counter+1
+    with open('data.json', 'w') as f:
+        json.dump(para_dic, f)
+    redirect_url = request.url_for('training')
     return RedirectResponse(redirect_url, status_code=303)
 
 
 
 
 
+
     #return redirect(url_for('delete_images'))
     # redirect_url = request.url_for('basic_upload')
     # return RedirectResponse(redirect_url, status_code=303)