Changed purpose of app.py to create gradio.Blocks interface

Explanation of the Error:

#Nested Lists Not Allowed: Gradio's Interface does not support nested lists within inputs or outputs. Each element should be a single component (e.g., gr.Textbox, gr.Image), not a list of components.
#Multiple Functions Handling: When using multiple functions with different input/output configurations, the Interface class alone isn't sufficient.

Solution: Use Gradio's Blocks API

#To build a more complex interface that can handle multiple functions with varying numbers of inputs and outputs, you should use Gradio's Blocks API. This allows you to create a custom layout with multiple components and interactions.

Explanation of the Changes:

#Used gr.Blocks: This allows for a more flexible layout and supports complex interactions.
#Organized into Tabs:

Tab 1: "Correct HTR" with input raw_htr_input and output corrected_output.
Tab 2: "Summarize Legal Text" with input legal_text_input and output summary_output.
Tab 3: "Answer Legal Question" with inputs legal_text_input_q and question_input, and output answer_output.
Added Buttons: Each function is triggered by a button click, which calls the corresponding function.

app.py

@@ -1,3 +1,4 @@
+import os
 import gradio as gr
 from transformers import T5Tokenizer, T5ForConditionalGeneration
 
@@ -36,22 +37,32 @@ def answer_question(legal_text, question):
 
     return answer
 
-# Create the Gradio interface
-iface = gr.Interface(
-    fn=[correct_htr, summarize_text, answer_question],
-    inputs=[
-        gr.Textbox(lines=5, placeholder="Enter raw HTR text here..."),
-        gr.Textbox(lines=10, placeholder="Enter legal text to summarize..."),
-        [gr.Textbox(lines=10, placeholder="Enter legal text..."),
-         gr.Textbox(lines=2, placeholder="Enter your question...")]
-    ],
-    outputs=[
-        gr.Textbox(lines=5, placeholder="Corrected HTR text"),
-        gr.Textbox(lines=5, placeholder="Summary of legal text"),
-        gr.Textbox(lines=5, placeholder="Answer to your question")
-    ],
-    title="mT5 Legal Assistant",
-    description="Use this tool to correct raw HTR, summarize legal texts, or answer questions about legal cases."
-)
-
-iface.launch()
+# Create the Gradio Blocks interface
+with gr.Blocks() as demo:
+    gr.Markdown("# mT5 Legal Assistant")
+    gr.Markdown("Use this tool to correct raw HTR, summarize legal texts, or answer questions about legal cases.")
+
+    with gr.Tab("Correct HTR"):
+        gr.Markdown("### Correct Raw HTR Text")
+        raw_htr_input = gr.Textbox(lines=5, placeholder="Enter raw HTR text here...")
+        corrected_output = gr.Textbox(lines=5, placeholder="Corrected HTR text")
+        correct_button = gr.Button("Correct HTR")
+        correct_button.click(correct_htr, inputs=raw_htr_input, outputs=corrected_output)
+
+    with gr.Tab("Summarize Legal Text"):
+        gr.Markdown("### Summarize Legal Text")
+        legal_text_input = gr.Textbox(lines=10, placeholder="Enter legal text to summarize...")
+        summary_output = gr.Textbox(lines=5, placeholder="Summary of legal text")
+        summarize_button = gr.Button("Summarize Text")
+        summarize_button.click(summarize_text, inputs=legal_text_input, outputs=summary_output)
+
+    with gr.Tab("Answer Legal Question"):
+        gr.Markdown("### Answer a Question Based on Legal Text")
+        legal_text_input_q = gr.Textbox(lines=10, placeholder="Enter legal text...")
+        question_input = gr.Textbox(lines=2, placeholder="Enter your question...")
+        answer_output = gr.Textbox(lines=5, placeholder="Answer to your question")
+        answer_button = gr.Button("Get Answer")
+        answer_button.click(answer_question, inputs=[legal_text_input_q, question_input], outputs=answer_output)
+
+demo.launch()
+