Update app.py

app.py

@@ -9,30 +9,39 @@ from langchain.chains import LLMChain
 openai.api_key = os.environ.get("OPENAI_API_KEY")
 
 # Prompt Template
-prompt = PromptTemplate(
-    input_variables=["student_input", "tone"],
+# Prompt template
+summarize_Prompt = PromptTemplate(
+    input_variables=["student_input"],
+    template="Summarize the following writing:\\n{student_input}"
+)
+feedback_prompt = PromptTemplate(
+    input_variables=["summary", "tone"],
     template="""You are a writing coach who gives {tone} feedback.
-Here is a student's answer: {student_input}
-Respond with 2-3 sentences of constructive, supportive advice.
+Here's a summary of a student's writing: {summary}
+Give 2-3 sentences of constructive suggestions to improve the writing.
 """
 )
 
+
 # Update these imports to use the new recommended packages
 from langchain_community.chat_models import ChatOpenAI  # or from langchain_openai import ChatOpenAI
 
 llm = ChatOpenAI(model="gpt-3.5-turbo", temperature=0.7)
-chain = LLMChain(llm=llm, prompt=prompt)
+summarize_chain = LLMChain(llm=llm, prompt=summarize_Prompt)
+feedback_chain = LLMChain(llm=llm, prompt=feedback_prompt)
 
-def generate_feedback(student_input, tone="supportive"):
+
+# Feedback function using langChain
+def generate_feedback(student_input, tone='supportive'):
     try:
-        return chain.run({
-            "student_input": student_input,
-            "tone": tone
-        })
+        summary = summarize_chain.run({'student_input': student_input})
+        final_feedback = feedback_chain.run({'summary': summary, 'tone': tone})
+        return final_feedback
     except Exception as e:
-        print("Error during Langchain call:", e)
         return f"Error: {str(e)}"
 
+        
+
 # Build the UI
 with gr.Blocks() as demo:
     gr.Markdown("#0  AI Feedback System\nChoose a tone and get feedback on your writing.")