Update app.py

app.py

@@ -4,12 +4,12 @@ from groq import Groq
 from io import BytesIO
 from PyPDF2 import PdfReader
 import docx
-#import json 
 
 # Initialize Groq client
 client = Groq(api_key=os.environ.get("Groq_Api_Key"))
 
-# Function to extract text from PDF
+# Helper Functions
+
 def extract_text_from_pdf(pdf_file):
     pdf_reader = PdfReader(pdf_file)
     text = ""
@@ -17,14 +17,11 @@ def extract_text_from_pdf(pdf_file):
         text += page.extract_text()
     return text
 
-# Function to extract text from Word document
 def extract_text_from_word(doc_file):
     doc = docx.Document(doc_file)
     text = "\n".join([paragraph.text for paragraph in doc.paragraphs])
     return text
 
-# Helper Functions
-
 def summarize_topic(text, topic):
     messages = [
         {"role": "system", "content": "Summarize the following lesson content."},
@@ -48,7 +45,6 @@ def generate_mcqs(text, num_questions):
     ]
     response = client.chat.completions.create(messages=messages, model="llama-3.3-70b-versatile", stream=False)
     return eval(response.choices[0].message.content)
-    
 
 def adapt_lesson_for_grade(text, grade):
     messages = [
@@ -58,77 +54,130 @@ 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
 
-# Streamlit app layout
-st.title("AI 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.
-""")
+# Streamlit App Layout
+st.set_page_config(
+    page_title="EduAI Assistant",
+    page_icon="\ud83d\udcda",
+    layout="wide",
+)
+
+# Header Section with Branding
+st.markdown(
+    """
+    <style>
+        .main-header {
+            text-align: center;
+            background-color: #f4f4f4;
+            padding: 15px;
+            border-radius: 5px;
+        }
+    </style>
+    <div class="main-header">
+        <h1>EduAI - Your Teaching Assistant</h1>
+        <p>AI-powered tools to assist educators with lesson planning, quizzes, and content adaptation.</p>
+    </div>
+    """,
+    unsafe_allow_html=True,
+)
 
 # Sidebar: File Upload and Options
-st.sidebar.header("Upload Files or Enter Text")
-uploaded_files = st.sidebar.file_uploader("Upload lesson files (PDFs or Word documents)", accept_multiple_files=True)
-manual_input = st.sidebar.text_area("Or paste lesson text here", height=200)
-
-# Select Functionality
-st.sidebar.header("Select Action")
-task = st.sidebar.selectbox("What would you like to do?", [
-    "Summarize a Topic",
-    "Ask Questions",
-    "Generate MCQs",
-    "Adapt Lesson for Grades"
-])
-
-# Main Actions
-if manual_input or uploaded_files:
-    # Combine uploaded files into a single text
-    combined_text = ""
-    if uploaded_files:
-        for file in uploaded_files:
-            file_type = file.name.split(".")[-1].lower()
-            if file_type == "pdf":
-                combined_text += extract_text_from_pdf(file)
-            elif file_type == "docx":
-                combined_text += extract_text_from_word(file)
-            else:
-                st.error(f"Unsupported file type: {file_type}")
-                st.stop()
-
-    lesson_text = combined_text if uploaded_files else manual_input
-
-    if task == "Summarize a Topic":
-        topic = st.text_input("Enter the topic or keywords:")
-        if st.button("Summarize"):
-            summary = summarize_topic(lesson_text, topic)
-            st.write("### Summary")
-            st.write(summary)
-
-    elif task == "Ask Questions":
-        question = st.text_input("Enter your question:")
-        if st.button("Get Answer"):
-            answer = ask_question(lesson_text, question)
-            st.write("### Answer")
-            st.write(answer)
-
-    elif task == "Generate MCQs":
-        num_questions = st.slider("Number of questions to generate:", 1, 10, 5)
-        if st.button("Generate MCQs"):
-            mcqs = generate_mcqs(lesson_text, num_questions)
-            st.write("### Multiple Choice Questions")
-            for i, mcq in enumerate(mcqs, 1):
-                st.write(f"**Q{i}. {mcq['question']}**")
-                for option in mcq['options']:
-                    st.write(f"- {option}")
-                st.write(f"**Answer:** {mcq['answer']}")
-                st.write("---")
-
-    elif task == "Adapt Lesson for Grades":
-        grade = st.selectbox("Select the target grade level:", ["Grade 1", "Grade 5", "Grade 8", "Grade 12"])
-        if st.button("Adapt Lesson"):
-            adapted_lesson = adapt_lesson_for_grade(lesson_text, grade)
-            st.write(f"### Lesson Adapted for {grade}")
-            st.write(adapted_lesson)
-
-else:
-    st.info("Please upload files or enter lesson text to begin.")
+with st.sidebar:
+    st.header("\ud83c\udf93 Upload Lessons or Enter Text")
+    uploaded_files = st.file_uploader("Upload lesson files (PDFs or Word documents)", accept_multiple_files=True)
+    manual_input = st.text_area("Or paste lesson text here", height=200)
+
+    st.header("\u2699\ufe0f Choose Action")
+    task = st.selectbox("What would you like to do?", [
+        "Summarize a Topic",
+        "Ask Questions",
+        "Generate MCQs",
+        "Adapt Lesson for Grades"
+    ])
+
+    st.header("\ud83d\udd04 Additional Options")
+    st.checkbox("Show raw outputs", key="show_raw_outputs")
+
+# Main Layout with Tabs
+tabs = st.tabs(["\ud83d\udcc2 Lesson Input", "\ud83d\udcdc Outputs", "\ud83d\udd8b\ufe0f Help & Info"])
+
+# Lesson Input Tab
+with tabs[0]:
+    st.subheader("Lesson Text Preview")
+    if manual_input or uploaded_files:
+        combined_text = ""
+        if uploaded_files:
+            for file in uploaded_files:
+                file_type = file.name.split(".")[-1].lower()
+                try:
+                    if file_type == "pdf":
+                        combined_text += extract_text_from_pdf(file)
+                    elif file_type == "docx":
+                        combined_text += extract_text_from_word(file)
+                    else:
+                        st.error(f"Unsupported file type: {file_type}")
+                        st.stop()
+                except Exception as e:
+                    st.error(f"Error reading file {file.name}: {e}")
+        
+        lesson_text = combined_text if uploaded_files else manual_input
+        st.text_area("Lesson Content:", value=lesson_text, height=300)
+    else:
+        st.info("Please upload files or enter lesson text to begin.")
+
+# Outputs Tab
+with tabs[1]:
+    st.subheader("Results")
+    if manual_input or uploaded_files:
+        if task == "Summarize a Topic":
+            topic = st.text_input("Enter the topic or keywords:")
+            if st.button("Summarize"):
+                with st.spinner("Summarizing..."):
+                    summary = summarize_topic(lesson_text, topic)
+                st.success("Summary generated!")
+                st.write("### Summary")
+                st.write(summary)
+
+        elif task == "Ask Questions":
+            question = st.text_input("Enter your question:")
+            if st.button("Get Answer"):
+                with st.spinner("Finding answer..."):
+                    answer = ask_question(lesson_text, question)
+                st.success("Answer retrieved!")
+                st.write("### Answer")
+                st.write(answer)
+
+        elif task == "Generate MCQs":
+            num_questions = st.slider("Number of questions to generate:", 1, 10, 5)
+            if st.button("Generate MCQs"):
+                with st.spinner("Generating MCQs..."):
+                    mcqs = generate_mcqs(lesson_text, num_questions)
+                st.success("MCQs generated!")
+                for i, mcq in enumerate(mcqs, 1):
+                    st.write(f"**Q{i}. {mcq['question']}**")
+                    for option in mcq['options']:
+                        st.write(f"- {option}")
+                    st.write(f"**Answer:** {mcq['answer']}")
+                    st.write("---")
+
+        elif task == "Adapt Lesson for Grades":
+            grade = st.selectbox("Select the target grade level:", ["Grade 1", "Grade 5", "Grade 8", "Grade 12"])
+            if st.button("Adapt Lesson"):
+                with st.spinner("Adapting lesson..."):
+                    adapted_lesson = adapt_lesson_for_grade(lesson_text, grade)
+                st.success(f"Lesson adapted for {grade}!")
+                st.write(f"### Lesson Adapted for {grade}")
+                st.write(adapted_lesson)
+    else:
+        st.info("No text or files uploaded. Please provide input.")
+
+# Help & Info Tab
+with tabs[2]:
+    st.subheader("How to Use EduAI")
+    st.markdown(
+        """
+        - **Upload Files**: Drag and drop your PDF or Word documents.
+        - **Choose Action**: Select the task you want EduAI to perform.
+        - **View Outputs**: Check the results under the Outputs tab.
+        - **Contact Support**: For any issues, contact [support@eduai.com](mailto:support@eduai.com).
+        """
+    )