Update app.py

app.py

@@ -139,12 +139,23 @@ def adapt_lesson_for_grade(text, grade):
     response = client.chat.completions.create(messages=messages, model="llama-3.3-70b-versatile", stream=False)
     return response.choices[0].message.content
 
+def generate_assignment(text, topic):
+    """
+    Generate a conceptual-based assignment from the given lesson or topic.
+    """
+    messages = [
+        {"role": "system", "content": "Generate a conceptual-based assignment from the following lesson content."},
+        {"role": "user", "content": f"Context: {text}\n\nGenerate a conceptual-based assignment for the topic: {topic}."},
+    ]
+    response = client.chat.completions.create(messages=messages, model="llama-3.3-70b-versatile", stream=False)
+    return response.choices[0].message.content
+
 # Streamlit app layout
 st.title("EduAI Assistant for Teachers")
 st.markdown("""
 Welcome to the AI-powered teaching assistant!  
 - Upload your lesson files or input text.  
-- Ask questions, summarize topics, or create quizzes and assignments tailored to different grades.
+- Ask questions, summarize topics, create quizzes and assignments tailored to different grades.
 """)
 
 # Sidebar: File Upload and Options
@@ -158,7 +169,8 @@ task = st.sidebar.selectbox("What would you like to do?", [
     "Summarize a Topic",
     "Ask Questions",
     "Generate MCQs",
-    "Adapt Lesson for Grades"
+    "Adapt Lesson for Grades",
+    "Generate Conceptual Assignment"  # Added new option for assignment generation
 ])
 
 # Main Actions
@@ -214,5 +226,12 @@ if manual_input or uploaded_files:
             st.write(f"### Lesson Adapted for {grade}")
             st.write("\n\n".join(adapted_lessons))
 
+    elif task == "Generate Conceptual Assignment":
+        topic = st.text_input("Enter the topic for assignment generation:")
+        if st.button("Generate Assignment"):
+            assignments = [generate_assignment(chunk, topic) for chunk in text_chunks]
+            st.write("### Conceptual Assignment")
+            st.write("\n\n".join(assignments))
+
 else:
     st.info("Please upload files or enter lesson text to begin.")