Update app.py

app.py

@@ -133,7 +133,7 @@ class MistralChatBot:
     def __init__(self, system_prompt="Le dialogue suivant est une conversation"):
         self.system_prompt = system_prompt
 
-    def predict_field(self, user_message, context):
+    def predict_field(self, user_message):
         detailed_prompt = """### Question ###\n""" + user_message + "\n\n### Field ###\n"
         prompts = [detailed_prompt]
         outputs = llm.generate(prompts, sampling_params, use_tqdm = False)
@@ -184,8 +184,10 @@ with gr.Blocks(theme='gradio/monochrome') as demo:
     text_input = gr.Textbox(label="Your question", type="text", lines=1)
     context_input = gr.Textbox(label="Your context", type="text", lines=1)
     text_button = gr.Button("Query InaguaLLM")
+    field_output = gr.HTML(label="Field")
     text_output = gr.HTML(label="Answer")
-    text_button.click(mistral_bot.predict, inputs=[text_input, context_input], outputs=[text_output], api_name="convert-code")
+    text_button.click(mistral_bot.predict_field, inputs=[text_input], outputs=[field_output], api_name="convert-question")
+    text_button.click(mistral_bot.predict_answer, inputs=[text_input, context_input], outputs=[text_output], api_name="convert-code")
 
 if __name__ == "__main__":
     demo.queue().launch()