Update app.py

app.py

@@ -1,8 +1,59 @@
 import gradio as gr
 
+from utils.state import State
+from utils.process_input import get_response
+from utils.examples import examples
+
+
+def clear_fields():
+    app_state.clear_all()
+    return None, None, None
+
+
+def generate_response(instructions, input_mail):
+    try:
+        model = app_state.model
+        response = get_response(
+            model=model, instructions=instructions, input_texts=input_mail
+        )
+
+        return response
+    except Exception as e:
+        gr.Error(e)
+
+
 with gr.Blocks() as demo:
-    with gr.Column():
-        gr.TextArea()
+    gr.Markdown("# Phi3 Mini 4k Model Fine-tuned")
+    gr.Markdown(
+        "**NOTE**: Please note that the response generated by the model might not be accurate. Always verify results."
+    )
+    with gr.Row(equal_height=True):
+        with gr.Column():
+            instructions = gr.TextArea(label="System Instructions", max_lines=2)
+            input_mail = gr.TextArea(label="Input Email", max_lines=6)
+            with gr.Row():
+                submit_button = gr.Button("Submit", variant="primary")
+                clear_button = gr.Button("Clear")
+
+        with gr.Column():
+            with gr.Row():
+                extracted_fields = gr.TextArea(label="Extracted Fields")
+
+    gr.Examples(examples=examples, inputs=[instructions, input_mail])
+
+    clear_button.click(
+        fn=clear_fields,
+        outputs=[instructions, input_mail, extracted_fields],
+    )
+    submit_button.click(
+        fn=generate_response,
+        inputs=[instructions, input_mail],
+        outputs=[extracted_fields],
+    )
 
 if __name__ == "__main__":
-    demo.launch()
+    try:
+        app_state = State()
+        demo.launch()
+    except Exception as e:
+        gr.Error(e)