Enter code, show results GUI works.

eval_code.py

@@ -41,7 +41,7 @@ def eval_code_by_chatgpt(openai_client, ccode):
     :param ccode:
     :return:
     """
-    time.sleep(3)
+    # time.sleep(3)
     try:
         return """[
   {

main.py

@@ -23,7 +23,8 @@ session_state_table = global_database_tables.session_state
 if session_state_table not in global_database_tables:
     # session_id stored in cookies
     # see EVAL_STATUS_x below for state
-    session_state_table.create(id=int, session_id=str, state=int, submitted=datetime, completed=datetime, answer=str, pk='id')
+    session_state_table.create(id=int, session_id=str, state=int, submitted=datetime,
+                               completed=datetime, answer=str, code=str, pk='id')
 Session_State_cls = session_state_table.dataclass()
 
 EVAL_STATE_NEW=0
@@ -32,14 +33,21 @@ EVAL_STATE_TIMEDOUT=2
 EVAL_STATE_ANSWER=3
 EVAL_STATE_ERROR=4
 
+# Constants to name the various HTML ids in the code
+HTML_SUBMIT_CODE_AREA = "submit_code_area"
+HTML_RESULTS_AREA = "prompt_response"
+HTML_CLEAR_FORM = "clear_the_form"
+
+hdrs = (HighlightJS(langs=['python', 'javascript', 'html', 'css']),)
 
 if "localhost" in SPACE_HOST:
     # Are we hacking locally?
     print("Localhost detected in SPACE_HOST. App started in debug+live mode!")
-    app, rt = fast_app(debug=True, live=True)
+    app, rt = fast_app(debug=True, live=True, hdrs=hdrs)
+    REFRESH_TIME = 0.1
 else:
-    app, rt = fast_app(debug=False, live=False)
-
+    app, rt = fast_app(debug=False, live=False, hdrs=hdrs)
+    REFRESH_TIME = 1
 
 CODE_AUGMENTATIONS=[
     ("DRY", "Don't repeat yourself."),
@@ -48,32 +56,61 @@ CODE_AUGMENTATIONS=[
     ("NAME", "Meaningful names in the code."),
 ]
 
-def format_code_review(code_js, html_id=""):
+def format_code_review(feedback_js, c_code, html_id=""):
     list_of_citerias = []
     for caug_code, caug_txt in CODE_AUGMENTATIONS:
         crit_tag = [H3(caug_code), P(caug_txt)]
         list_of_citerias.extend(crit_tag)
         # yeah, I know . . .
         criteria_found = False
-        for eval_line in code_js:
+        for eval_line in feedback_js:
             if caug_code == eval_line["criteria"]:
-                eval_txt = Pre(eval_line["explanation"])
+                eval_txt = P(eval_line["explanation"])
                 list_of_citerias.append(eval_txt)
                 criteria_found = True
         if criteria_found is False:
             list_of_citerias.append(P("Not infringed"))
-    return Div(*list_of_citerias, _id=html_id)
+    return Div(html_render_code_output(c_code), *list_of_citerias, _id=html_id)
 
-def generate_default_eval_section(html_id=""):
+def html_default_results(html_id):
     return Div(P("This is where criterias will show up once the code is evaluated"), _id=html_id)
 
-def generate_working_eval_section(html_id=""):
+def html_waiting_for_results(html_id):
     return Div(P("Working  . . ."), _id=html_id,
     hx_get=f"/render_answer",
-    hx_trigger = "every 2s",
+    hx_trigger = f"every {REFRESH_TIME}s",
     hx_swap = "outerHTML",
     )
 
+def get_latest_eval_request_status(session_id):
+    state_rows = session_state_table(limit=1, where=f"session_id == '{session_id}'", order_by="id DESC")
+    if len(state_rows) <= 0:
+        return EVAL_STATE_NEW, None
+    state_obj = state_rows[0]
+    if state_obj.state in {EVAL_STATE_NEW, EVAL_STATE_QUERY, EVAL_STATE_ANSWER}:
+        return  state_obj.state, state_obj
+    return EVAL_STATE_ERROR, state_obj
+
+
+def html_error_results(message, html_id):
+    div = Div(P("There was an error:", P(message)), _id=html_id)
+    return div
+
+def html_render_answer_from_db(session_id, html_id):
+    eval_request_status, state_obj = get_latest_eval_request_status(session_id)
+    # state_rows = session_state_table(limit=1, where=f"session_id == '{session_id}'", order_by="id DESC")
+    # print(eval_request_status, state_obj)
+    if eval_request_status == EVAL_STATE_NEW:
+        return html_default_results(html_id), #, html_render_inputbox(HTML_RESULTS_AREA, HTML_SUBMIT_CODE_AREA)
+    if eval_request_status == EVAL_STATE_ANSWER:
+        gpt_js_eval = json.loads(state_obj.answer)
+        return (format_code_review(gpt_js_eval, state_obj.code, html_id),
+                html_render_inputbox(target_html_id=HTML_RESULTS_AREA, region_html_id=HTML_SUBMIT_CODE_AREA)) #, html_render_code_output(HTML_SUBMIT_CODE_AREA, state_obj.code)
+    if eval_request_status == EVAL_STATE_QUERY:
+        return html_waiting_for_results(html_id),
+    return html_error_results(state_obj.answer, html_id), #, html_render_code_output(HTML_SUBMIT_CODE_AREA, state_obj.code)
+
+
 # How can I timeout? Well ... TBD.
 @threaded
 def call_gpt_and_store_result(pkid, code_to_check):
@@ -102,36 +139,68 @@ def call_gpt_and_store_result(pkid, code_to_check):
         import traceback
         traceback.print_exc()
 
-def get_rendered_answer(session_id):
-    state_rows = session_state_table(limit=1, where=f"session_id == '{session_id}'", order_by="id DESC")
-    # print(state_rows)
-    if len(state_rows) <= 0:
-        return generate_default_eval_section(html_id="prompt_response")
-    state = state_rows[0]
-    if state.state == EVAL_STATE_ANSWER:
-        gpt_js_eval = json.loads(state.answer)
-        return format_code_review(gpt_js_eval, html_id="prompt_response")
-    if state.state == EVAL_STATE_QUERY:
-        return generate_working_eval_section(html_id="prompt_response")
-    return Div(P("There was an error:", P(state.answer)), _id="prompt_response")
 
 
-@rt("/render_answer")
-def get(session):
-    if 'session_id' not in session: return "No session ID"
-    session_id = session["session_id"]
-    return get_rendered_answer(session_id)
+def html_render_inputbox(target_html_id, region_html_id):
+    txtarea = Textarea(id="ccodetoeval", name="ccodetoeval", placeholder="Enter a piece of C code", rows=3)
+    form = Form(Group(txtarea, Button("Evaluate")),
+                hx_post="/submit_to_eval",
+                hx_swap="outerHTML",
+                target_id=target_html_id
+                )
+    return Div(form, _id=region_html_id, hx_swap_oob='true')
+
+def html_render_code_output(code):
+    txtarea = Pre(Code(code))
+    return txtarea
+
+
+def render_conditional_inputbox_results(session_id):
+    eval_request_status, _ = get_latest_eval_request_status(session_id)
+    pass
+
+def html_render_clear_area_button(html_id):
+    button = Button("Clear form",
+                    hx_get="/clear_area",
+                    hx_swap="outerHTML",
+                    target_id=HTML_RESULTS_AREA,
+                    )
+    div = Div(button, _id=html_id, hx_swap_oob='true')
+    return div
+
+def render_clear_area(session_id, html_id):
+    # return html_render_clear_area_button(html_id)
+    eval_request_status, _ = get_latest_eval_request_status(session_id)
+    if eval_request_status != EVAL_STATE_NEW:
+        print("clear button: render button")
+        return html_render_clear_area_button(html_id)
+    else:
+        print("clear button: render empty")
+        return Div(P(""), _id=html_id, hx_swap_oob='true')
+
 
 @rt("/")
 def get(session):
     if 'session_id' not in session:
         session['session_id'] = str(uuid.uuid4())
-    inp = Textarea(id="ccodetoeval", name="ccodetoeval", placeholder="Enter a piece of C code", rows=20)
-    # eval_area = Pre("", id="prompt_response")
-    # eval_area = format_code_review([], html_id="prompt_response")
-    eval_area = generate_default_eval_section(html_id="prompt_response")
-    add = Form(Group(inp, Button("Generate")), hx_post="/submit_to_eval", hx_swap="outerHTML",  target_id='prompt_response')
-    return Title('Image Generation Demo'), Main(H1('Magic Image Generation'), add, eval_area, cls='container')
+    session_id = session["session_id"]
+    title = Title('C code review for students')
+    preamble = [H1("Evaluate your C code!"),
+                P("Enter your code in the textbox below and wait for answers."),
+                P("!! The data will be saved and maybe made public !!"),
+                ]
+
+    input_area = html_render_inputbox(target_html_id=HTML_RESULTS_AREA, region_html_id=HTML_SUBMIT_CODE_AREA)
+    results_area = html_render_answer_from_db(session_id, HTML_RESULTS_AREA)
+    clear_area = render_clear_area(session_id, HTML_CLEAR_FORM)
+    return title, Main(*preamble, input_area, results_area, clear_area)
+
+@rt("/render_answer")
+def get(session):
+    if 'session_id' not in session: return "No session ID"
+    session_id = session["session_id"]
+    answer_area = html_render_answer_from_db(session_id, HTML_RESULTS_AREA)
+    return answer_area
 
 
 @rt("/submit_to_eval", methods="post")
@@ -142,13 +211,37 @@ def post(ccodetoeval:str, session):
         session_id=session_id,
         state=EVAL_STATE_QUERY,
         submitted=datetime.utcnow(),
+        code=ccodetoeval,
         answer=""
     )
     # we insert and we get the new primary key
     session_obj = session_state_table.insert(session_obj)
     # will be executed in another thread with magic @threaded
     call_gpt_and_store_result(session_obj.id, ccodetoeval)
-    return get_rendered_answer(session_id)
+    return (*html_render_answer_from_db(session_id, HTML_RESULTS_AREA),
+            render_clear_area(session_id, HTML_CLEAR_FORM)
+            )
+
+
+@rt("/clear_area", methods="get")
+def get(session):
+    if 'session_id' not in session: return P("Bad call. No session ID")
+    session_id = session["session_id"]
+    # insert a row to "cancel"/reset the current request
+    session_obj = Session_State_cls(
+        session_id=session_id,
+        state=EVAL_STATE_NEW,
+        submitted=datetime.utcnow(),
+        code="",
+        answer=""
+    )
+    session_state_table.insert(session_obj)
+    # re-issue forms
+    input_area = html_render_inputbox(target_html_id=HTML_RESULTS_AREA, region_html_id=HTML_SUBMIT_CODE_AREA)
+    results_area = html_render_answer_from_db(session_id, HTML_RESULTS_AREA)
+    clear_area = render_clear_area(session_id, HTML_CLEAR_FORM)
+    print(results_area)
+    return *results_area, input_area, clear_area
 
 
 ## This code is for reference, for OAuth. BIG PITA, will be added later.