Update app.py

app.py

@@ -1,6 +1,9 @@
+import gspread
+from oauth2client.service_account import ServiceAccountCredentials
 import streamlit as st
 import openai
 import time
+from datetime import datetime
 
 # Set your OpenAI API key from Hugging Face Secrets
 openai.api_key = st.secrets["OPENAI_API_KEY"]
@@ -8,6 +11,33 @@ openai.api_key = st.secrets["OPENAI_API_KEY"]
 # Initialize OpenAI client
 client = openai.OpenAI(api_key=openai.api_key)
 
+# Google Sheets setup
+scope = ["https://spreadsheets.google.com/feeds", "https://www.googleapis.com/auth/drive"]
+creds = ServiceAccountCredentials.from_json_keyfile_name("path/to/credentials.json", scope)
+client_gs = gspread.authorize(creds)
+sheet = client_gs.open("API_Usage").sheet1  # Open the Google Sheet
+
+# Function to get user data from Google Sheets
+def get_user_data(user_id):
+    users = sheet.get_all_records()
+    for user in users:
+        if user['UserID'] == user_id:
+            return user
+    return None
+
+# Function to update user data in Google Sheets
+def update_user_data(user_id, daily_api_count, total_api_count, last_used_date):
+    cell = sheet.find(user_id)
+    if cell:
+        sheet.update_cell(cell.row, cell.col + 1, daily_api_count)  # Update DailyAPICount
+        sheet.update_cell(cell.row, cell.col + 2, total_api_count)  # Update TotalAPICount
+        sheet.update_cell(cell.row, cell.col + 3, last_used_date)   # Update LastUsedDate
+    else:
+        sheet.append_row([user_id, daily_api_count, total_api_count, last_used_date])  # Add new user if not found
+
+# Admin user ID
+admin_user_id = "admin"
+
 # Function to generate exam questions using OpenAI API with retry logic
 def generate_questions_with_retry(knowledge_material, question_type, cognitive_level, extra_instructions, case_based, num_choices=None, max_retries=3):
     # Adjust the number of questions based on the type
@@ -40,7 +70,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}
@@ -54,7 +84,7 @@ def generate_questions_with_retry(knowledge_material, question_type, cognitive_l
                 st.error("Failed to connect to OpenAI API after several attempts.")
                 return None
 
-# Login page
+# Main logic for the app
 if 'username' not in st.session_state:
     # Show the login form if the username is not set
     st.title("Login")
@@ -66,108 +96,142 @@ if 'username' not in st.session_state:
         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) with 3,000-word limit
-    knowledge_material = st.text_area("Enter knowledge material to generate exam questions:")
-    
-    # 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")
+    user_id = st.session_state['username']
+    user_data = get_user_data(user_id)
     
-    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.")
+    if not user_data:
+        st.error("Invalid User ID. Please contact the admin for access.")
+    else:
+        # Get today's date
+        today = datetime.now().strftime("%Y-%m-%d")
+        
+        # Reset daily API count if it's a new day
+        if user_data['LastUsedDate'] != today:
+            update_user_data(user_id, 0, user_data['TotalAPICount'], today)
+            user_data['DailyAPICount'] = 0
+
+        # Check if the user has reached their daily limit (except for admin)
+        if user_id != admin_user_id and user_data['DailyAPICount'] >= 5:
+            st.warning("You have reached your daily limit of 5 API calls.")
         else:
-            # Here you can add code to extract text from the PDF if needed
-            # For simplicity, we're focusing on the text input for now
-            st.success("File uploaded successfully! (Text extraction not implemented yet.)")
-
-    # Select question type
-    question_type = st.selectbox("Select question type:", 
-                                 ["Multiple Choice", "Fill in the Blank", "Open-ended", "True/False"])
+            # Main app for generating questions
+            st.title(f"Welcome, {user_id}! Generate your exam questions")
 
-    # For multiple choice, let users select the number of choices
-    num_choices = None
-    if question_type == "Multiple Choice":
-        num_choices = st.selectbox("Select the number of choices for each question:", [3, 4, 5])
-
-    # Select cognitive level
-    cognitive_level = st.selectbox("Select cognitive level:", 
-                                   ["Recall", "Understanding", "Application", "Analysis", "Synthesis", "Evaluation"])
-
-    # Checkbox for Case-Based Medical Situations
-    case_based = st.checkbox("Generate case-based medical exam questions")
-
-    # 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
-    if 'previous_questions' not in st.session_state:
-        st.session_state['previous_questions'] = []
-
-    if st.button("Generate Questions"):
-        if len(knowledge_material.split()) <= 3000:
-            # Generate questions with retry logic
-            questions = generate_questions_with_retry(
-                knowledge_material, 
-                question_type, 
-                cognitive_level, 
-                extra_instructions, 
-                case_based, 
-                num_choices
-            )
+            # Input field for knowledge material (text) with 3,000-word limit
+            knowledge_material = st.text_area("Enter knowledge material to generate exam questions:")
             
-            if questions:
-                st.write("Generated Exam Questions:")
-                st.write(questions)
-
-                # Avoid showing repeated content in future requests
-                st.session_state['previous_questions'].append(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 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 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, 
-                case_based, 
-                num_choices
-            )
+            # Word count check
+            if len(knowledge_material.split()) > 3000:
+                st.warning("Please limit the knowledge material to 3,000 words or fewer.")
 
-            # Check if the new set of questions is not the same as the previous set
-            if questions and questions not in st.session_state['previous_questions']:
-                st.write("Generated More Exam Questions:")
-                st.write(questions)
-
-                # Append the new questions to the session state
-                st.session_state['previous_questions'].append(questions)
-
-                # Option to download the new set of questions
-                st.download_button(
-                    label="Download More Questions",
-                    data=questions,
-                    file_name='more_generated_questions.txt',
-                    mime='text/plain'
-                )
-            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.")
+            # 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.")
+                else:
+                    st.success("File uploaded successfully! (Text extraction not implemented yet.)")
+
+            # Select question type
+            question_type = st.selectbox("Select question type:", 
+                                         ["Multiple Choice", "Fill in the Blank", "Open-ended", "True/False"])
+
+            # For multiple choice, let users select the number of choices
+            num_choices = None
+            if question_type == "Multiple Choice":
+                num_choices = st.selectbox("Select the number of choices for each question:", [3, 4, 5])
+
+            # Select cognitive level
+            cognitive_level = st.selectbox("Select cognitive level:", 
+                                           ["Recall", "Understanding", "Application", "Analysis", "Synthesis", "Evaluation"])
+
+            # Checkbox for Case-Based Medical Situations
+            case_based = st.checkbox("Generate case-based medical exam questions")
+
+            # 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
+            if 'previous_questions' not in st.session_state:
+                st.session_state['previous_questions'] = []
+
+            if st.button("Generate Questions"):
+                if len(knowledge_material.split()) <= 3000:
+                    # Generate questions with retry logic
+                    questions = generate_questions_with_retry(
+                        knowledge_material, 
+                        question_type, 
+                        cognitive_level, 
+                        extra_instructions, 
+                        case_based, 
+                        num_choices
+                    )
+                    
+                    if questions:
+                        st.write("Generated Exam Questions:")
+                        st.write(questions)
+
+                        # Avoid showing repeated content in future requests
+                        st.session_state['previous_questions'].append(questions)
+
+                        # Update user API usage
+                        new_daily_api_count = user_data['DailyAPICount'] + 1
+                        new_total_api_count = user_data['TotalAPICount'] + 1
+                        update_user_data(user_id, new_daily_api_count, new_total_api_count, today)
+                        
+                        st.success(f"API call #{new_daily_api_count} successful today!")
+                        st.info(f"Total API calls: {new_total_api_count}")
+
+                        # 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 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 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, 
+                        case_based, 
+                        num_choices
+                    )
+
+                    # Check if the new set of questions is not the same as the previous set
+                    if questions and questions not in st.session_state['previous_questions']:
+                        st.write("Generated More Exam Questions:")
+                        st.write(questions)
+
+                        # Append the new questions to the session state
+                        st.session_state['previous_questions'].append(questions)
+
+                        # Update user API usage for more questions
+                        new_daily_api_count = user_data['DailyAPICount'] + 1
+                        new_total_api_count = user_data['TotalAPICount'] + 1
+                        update_user_data(user_id, new_daily_api_count, new_total_api_count, today)
+
+                        st.success(f"API call #{new_daily_api_count} successful today!")
+                        st.info(f"Total API calls: {new_total_api_count}")
+
+                        # Option to download the new set of questions
+                        st.download_button(
+                            label="Download More Questions",
+                            data=questions,
+                            file_name='more_generated_questions.txt',
+                            mime='text/plain'
+                        )
+                    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.")
+
+