Update app.py

app.py

@@ -26,7 +26,7 @@ def generate_questions_with_retry(knowledge_material, question_type, cognitive_l
     while retries < max_retries:
         try:
             response = client.chat.completions.create(
-                model="gpt-4o-mini",
+                model="GPT-4o mini",
                 messages=[
                     {"role": "system", "content": "You are a helpful assistant for generating exam questions."},
                     {"role": "user", "content": prompt}
@@ -55,11 +55,19 @@ else:
     # Main App after login
     st.title(f"Welcome, {st.session_state['username']}! Generate your exam questions")
 
-    # Input field for knowledge material (text)
+    # Input field for knowledge material (text) with 3,000-word limit
     knowledge_material = st.text_area("Enter knowledge material to generate exam questions:")
-
-    # File uploader for PDFs (Optional: Only if you want to implement PDF input later)
+    
+    # Word count check
+    if len(knowledge_material.split()) > 3000:
+        st.warning("Please limit the knowledge material to 3,000 words or fewer.")
+    
+    # File uploader for PDFs (limited to 5 MB)
     uploaded_file = st.file_uploader("Upload a file (PDF)", type="pdf")
+    
+    if uploaded_file is not None:
+        if uploaded_file.size > 5 * 1024 * 1024:  # 5 MB limit
+            st.warning("File size exceeds 5 MB. Please upload a smaller file.")
 
     # Select question type
     question_type = st.selectbox("Select question type:", 
@@ -69,7 +77,7 @@ else:
     cognitive_level = st.selectbox("Select cognitive level:", 
                                    ["Recall", "Understanding", "Application", "Analysis", "Synthesis", "Evaluation"])
 
-    # Extra input field for additional instructions (now placed below cognitive level)
+    # Extra input field for additional instructions (placed below cognitive level)
     extra_instructions = st.text_area("Enter additional instructions (e.g., how you want the questions to be phrased):")
 
     # Generate questions button
@@ -77,7 +85,7 @@ else:
         st.session_state['previous_questions'] = []
 
     if st.button("Generate Questions"):
-        if knowledge_material:
+        if len(knowledge_material.split()) <= 3000:
             # Generate questions with retry logic
             questions = generate_questions_with_retry(knowledge_material, question_type, cognitive_level, extra_instructions)
             
@@ -96,11 +104,11 @@ else:
                     mime='text/plain'
                 )
         else:
-            st.warning("Please enter the knowledge material.")
+            st.warning("Please reduce the word count to 3,000 or fewer.")
 
     # Button to generate more questions based on the same material
     if st.button("Generate More Questions"):
-        if knowledge_material:
+        if len(knowledge_material.split()) <= 3000:
             # Regenerate new questions, trying to avoid repeated content
             questions = generate_questions_with_retry(knowledge_material, question_type, cognitive_level, extra_instructions)
 
@@ -121,3 +129,5 @@ else:
                 )
             else:
                 st.warning("New questions seem to overlap with the previous ones. Try adjusting the instructions.")
+        else:
+            st.warning("Please reduce the word count to 3,000 or fewer.")