Update app.py

app.py

@@ -42,6 +42,14 @@ prompt = PromptTemplate(input_variables=['task_type', 'task_number', 'question',
 # Define the LLM chain
 chain = LLMChain(llm=llm_model, prompt=prompt)
 
+# Define the function to toggle input visibility
+def toggle_input(input_type):
+    if input_type == "Image":
+        return gr.update(visible=True), gr.update(visible=False)
+    else:
+        return gr.update(visible=False), gr.update(visible=True)
+
+# Define the evaluation function
 def evaluate(task_type, task_number, question, input_type, image=None, text=None):
     if input_type == "Image" and image is not None:
         # Process the image to extract text
@@ -66,12 +74,7 @@ def evaluate(task_type, task_number, question, input_type, image=None, text=None
 
     return result
 
-# Function to toggle the visibility of inputs based on the selected input type
-def toggle_input(input_type):
-    if input_type == "Image":
-        return gr.update(visible=True), gr.update(visible=False)
-    else:
-        return gr.update(visible=False), gr.update(visible=True)
+
 
 footer_html_with_analytics = f"""
 <script async src="https://www.googletagmanager.com/gtag/js?id={tracking_id}"></script>
@@ -100,6 +103,7 @@ footer_html_with_analytics = f"""
 </div>
 """
 
+
 # Create the Gradio interface
 with gr.Blocks() as demo:
     gr.Markdown("# IELTS Writing Evaluation")
@@ -116,12 +120,15 @@ with gr.Blocks() as demo:
 
     input_type.change(toggle_input, [input_type], [image_input, text_input])
 
-    gr.Interface(
-        fn=evaluate, 
-        inputs=[test_type, task_number, question, input_type, image_input, text_input], 
-        outputs=gr.Markdown(label="Result")
-    )
-
+    result_output = gr.Markdown(label="Result")
+    
+    # Define the submit button and the function to be called
+    submit_button = gr.Button("Submit")
+    submit_button.click(fn=evaluate, 
+                        inputs=[test_type, task_number, question, input_type, image_input, text_input], 
+                        outputs=result_output)
+    
+    # Add the footer HTML
     gr.HTML(footer_html_with_analytics)
 
 # Launch the interface