submit function

app.py

@@ -73,23 +73,11 @@ class AASLDConversationalAgent():
 
 AASLD_CONVERSATIONAL_AGENT = AASLDConversationalAgent()
 
-PAGE_TITLE = """
-# Arithmedics AASLD Guidelines Assistant
-#### I'm a chatbot using gpt-4 model from OpenAI.
-### I help practictioners quickly get answers from AASLD Practice Guidelines
-"""
-
 PAGE_TITLE_HTML = """
 <h1> Arithmedics - AASLD Guidelines Assistant</h1> 
-<h4> I'm a chatbot using gpt-4 model from OpenAI. </h4>
 <h3> I help practictioners quickly get answers from AASLD Practice Guidelines </h3>
 """
 
-DEFAULT_ROLE = """
-You are a chat assistant who helps medical practictioners get quick answers from
-the practice guidelines published by "The American Association for the Study of 
-Liver Diseases (AASLD)." If you cannot find the answer please let the user know.
-"""
 def get_empty_state():
     return {'total_tokens': 0, 'messages': []}
 
@@ -97,9 +85,27 @@ def clear_conversation():
     return (gr.update(value=None, visible=True), None, '', get_empty_state(),
             '', '')
 
-def submit_question(question):
+def submit_question(question, messages):
     print(question)
-    return (question, '', '',  {'total_tokens': 0, 'messages': []}, '', '')
+    history = messages
+    prompt_msg = { 'role': 'user', 'content': question }
+    history.append(prompt_msg)
+    try:
+        answer = AASLD_CONVERSATIONAL_AGENT.get_answer(question)
+        print(answer)
+        answer_msg = { 'role': 'AI', 'content': answer }
+        history.append(answer_msg)
+    except Exception as e:
+        print(e)
+        history.append({
+            'role': 'system',
+            'content': f'Error: {e}'
+        })
+    chat_messages = [
+        (history[i]['content'], history[i+1]['content'])
+        for i in range(0, len(history)-1, 2)]
+    chat_messages.reverse()
+    return ('', chat_messages, '',  {'total_tokens': 0, 'messages': []}, question, answer)
 
 css = """
       #col-container {max-width: 80%; margin-left: auto; margin-right: auto;}
@@ -115,8 +121,6 @@ with gr.Blocks(css=css, title='AASLD Practice Guidelines ') as demo:
     state = gr.State(get_empty_state())
     with gr.Column(elem_id='col-container'):
         gr.Image('Arithmedics_rectangle_logo.png', elem_id='arithmedics-img')
-        # gr.Markdown(PAGE_TITLE,
-        #             elem_id='header')
         gr.HTML(PAGE_TITLE_HTML, elem_id='header')
         with gr.Row():
             with gr.Column(scale=7):