Update app.py

app.py

@@ -1,7 +1,7 @@
 import gradio as gr
 from transformers import pipeline
-from langchain import PromptTemplate
-from langchain.chains import LLMChain
+from langchain_core.prompts import PromptTemplate
+from langchain_core.runnables import RunnableSequence
 from langchain_google_genai import ChatGoogleGenerativeAI
 import os
 
@@ -95,10 +95,10 @@ Notes:
 
 - The text may have been obtained via OCR, which could result in some errors.
 - Disregard any text enclosed in <>. They separate different parts of the text.
-- If the Task type is empty, try to identify the task type from the question. If you cannot determine the task type, mention that the task type is unclear.
+- If the task type is empty, try to identify the task type from the question. If you cannot determine the task type, mention that the task type is unclear.
+- If the question section is empty, try to identify the question in the task content. 
 - If the specific question is not present in the task content, mention that the question does not exist in the task response.
 
-
 Task type: {task_type}
 
 Task number: {task_number}
@@ -114,11 +114,11 @@ llm_model = ChatGoogleGenerativeAI(model="gemini-pro", temperature=0.7, top_p=0.
 # Define the prompt template
 prompt = PromptTemplate(input_variables=['task_type', 'task_number', 'question', 'content', 'description'], template=initial_prompt)
 
-# Define the LLM chain
-chain = LLMChain(
-    llm=llm_model,
-    prompt=prompt,
-)
+# Define the RunnableSequence
+sequence = RunnableSequence([
+    prompt,
+    llm_model
+])
 
 def evaluate(task_type, task_number, question, image):
     # Process the image to extract text
@@ -128,8 +128,8 @@ def evaluate(task_type, task_number, question, image):
     # Select the appropriate description based on user input
     description = descriptions.get((task_type, task_number), "")
 
-    # Run the chain
-    result = chain.run({
+    # Run the sequence
+    result = sequence({
         'task_type': task_type,
         'task_number': task_number,
         'question': question,