Update app.py

app.py

@@ -44,11 +44,6 @@ chain = LLMChain(llm=llm_model, prompt=prompt)
 
 def evaluate(task_type, task_number, question, input_type, image=None, text=None):
     if input_type == "Image" and image is not None:
-        # Ensure the image is in the correct format
-        if isinstance(image, str):
-            # Load the image if it's a URL or path
-            image = Image.open(image)
-        
         # Process the image to extract text
         text_content = ocr_pipe(image)
         content = text_content[0]['generated_text']
@@ -71,16 +66,7 @@ def evaluate(task_type, task_number, question, input_type, image=None, text=None
 
     return result
 
-# Create the Gradio interface
-inputs = [
-    gr.Dropdown(choices=["Academic", "General"], label="Test Type", value="Academic"),
-    gr.Dropdown(choices=["Task 1", "Task 2"], label="Task Number", value="Task 1"),
-    gr.Textbox(label="Question", value=""),
-    gr.Radio(choices=["Image", "Text"], label="Input Type", value="Image"),
-    gr.Image(type="pil", label="Upload Image", visible=True),
-    gr.Textbox(label="Enter Text", visible=False)
-]
-
+# 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)
@@ -114,18 +100,28 @@ footer_html_with_analytics = f"""
 </div>
 """
 
-outputs = gr.Markdown(label="Result")
-
-# Define the Gradio Blocks and Interface
+# Create the Gradio interface
 with gr.Blocks() as demo:
     gr.Markdown("# IELTS Writing Evaluation")
+    
     with gr.Row():
-        with gr.Column():
-            input_type_radio = gr.Radio(choices=["Image", "Text"], label="Input Type", value="Image")
-            image_input = gr.Image(type="pil", label="Upload Image", visible=True)
-            text_input = gr.Textbox(label="Enter Text", visible=False)
-        input_type_radio.change(toggle_input, input_type_radio, [image_input, text_input])
-    gr.Interface(fn=evaluate, inputs=inputs, outputs=outputs)
+        test_type = gr.Dropdown(choices=["Academic", "General"], label="Test Type", value="Academic")
+        task_number = gr.Dropdown(choices=["Task 1", "Task 2"], label="Task Number", value="Task 1")
+        question = gr.Textbox(label="Question")
+        input_type = gr.Radio(choices=["Image", "Text"], label="Input Type", value="Image")
+        
+    with gr.Row():
+        image_input = gr.Image(type="pil", label="Upload Image", visible=True)
+        text_input = gr.Textbox(label="Enter Text", visible=False)
+
+    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")
+    )
+
     gr.HTML(footer_html_with_analytics)
 
 # Launch the interface