Update app.py

app.py

@@ -1,32 +1,70 @@
 import streamlit as st
 import openai
 
-# Set your OpenAI API key using Streamlit secrets
-openai.api_key = "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"
+# Set your OpenAI API key from Hugging Face Secrets
+openai.api_key = st.secrets["OPENAI_API_KEY"]
 
-# Create an OpenAI client using the new API structure
+# Initialize OpenAI client
 client = openai.OpenAI(api_key=openai.api_key)
 
-# Streamlit app
-st.title("Exam Question Generator")
-
-# Input field for user prompt
-user_input = st.text_area("Enter knowledge material to generate exam questions:")
-
-if st.button("Generate Questions"):
-    if user_input:
-        # Call OpenAI's Chat Completion API using the new structure
-        response = client.chat.completions.create(
-            model="gpt-3.5-turbo",  # Or use "gpt-4" if available
-            messages=[
-                {"role": "system", "content": "You are a helpful assistant for generating exam questions."},
-                {"role": "user", "content": f"Generate exam questions from the following material: {user_input}"}
-            ]
-        )
-
-        # Extract the response and display it correctly
-        st.write("Generated Exam Questions:")
-        st.write(response.choices[0].message.content)  # Access message content directly
-    else:
-        st.warning("Please enter the knowledge material.")
+# Function to generate exam questions using OpenAI API
+def generate_questions(knowledge_material, question_type, cognitive_level):
+    prompt = f"Generate {question_type.lower()} exam questions based on {cognitive_level.lower()} level from the following material: {knowledge_material}"
+    response = client.chat.completions.create(
+        model="gpt-4o-mini",  # You can use "gpt-4" if available
+        messages=[
+            {"role": "system", "content": "You are a helpful assistant for generating exam questions."},
+            {"role": "user", "content": prompt}
+        ]
+    )
+    return response.choices[0].message.content
+
+# Login page
+if 'username' not in st.session_state:
+    # Show the login form if the username is not set
+    st.title("Login")
+    username_input = st.text_input("Enter your username:")
+    if st.button("Login"):
+        if username_input:
+            st.session_state['username'] = username_input
+            st.success(f"Welcome, {username_input}!")
+        else:
+            st.warning("Please enter a valid username.")
+else:
+    # Main App after login
+    st.title(f"Welcome, {st.session_state['username']}! Generate your exam questions")
+
+    # Input field for knowledge material (text)
+    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)
+    uploaded_file = st.file_uploader("Upload a file (PDF)", type="pdf")
+
+    # Select question type
+    question_type = st.selectbox("Select question type:", 
+                                 ["Multiple Choice", "Fill in the Blank", "Open-ended"])
+
+    # Select cognitive level
+    cognitive_level = st.selectbox("Select cognitive level:", 
+                                   ["Recall", "Understanding", "Application", "Analysis", "Synthesis", "Evaluation"])
+
+    # Generate questions button
+    if st.button("Generate Questions"):
+        if knowledge_material:
+            # Generate questions using the OpenAI API
+            questions = generate_questions(knowledge_material, question_type, cognitive_level)
+
+            # Display the generated questions
+            st.write("Generated Exam Questions:")
+            st.write(questions)
+
+            # Option to download the questions as a text file
+            st.download_button(
+                label="Download Questions",
+                data=questions,
+                file_name='generated_questions.txt',
+                mime='text/plain'
+            )
+        else:
+            st.warning("Please enter the knowledge material.")