Update app.py

app.py

@@ -1,10 +1,6 @@
-import gspread
-from oauth2client.service_account import ServiceAccountCredentials
 import streamlit as st
 import openai
 import time
-from datetime import datetime
-from openai import OpenAIError  # Import OpenAIError for handling API errors
 
 # Set your OpenAI API key from Hugging Face Secrets
 openai.api_key = st.secrets["OPENAI_API_KEY"]
@@ -12,33 +8,6 @@ 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("genexam-2c8c645ecc0d.json", scope)
-client_gs = gspread.authorize(creds)
-sheet = client_gs.open("GeneXam user").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
@@ -70,32 +39,22 @@ def generate_questions_with_retry(knowledge_material, question_type, cognitive_l
     retries = 0
     while retries < max_retries:
         try:
-            # Use the new method `ChatCompletion.create`
-            response = openai.ChatCompletion.create(
-                model="GPT-4o mini",  # You can use "gpt-4" if available
+            response = client.chat.completions.create(
+                model="gpt-4o-mini",
                 messages=[
                     {"role": "system", "content": "You are a helpful assistant for generating exam questions."},
                     {"role": "user", "content": prompt}
-                ],
-                max_tokens=1000,
-                temperature=0.7
+                ]
             )
-            return response.choices[0].message["content"]  # Get the response text
-        except openai.APIConnectionError as e:
+            return response.choices[0].message.content
+        except openai.error.APIConnectionError:
             retries += 1
             time.sleep(2)  # Wait for 2 seconds before retrying
-            st.error(f"Failed to connect to OpenAI API: {str(e)}")
             if retries == max_retries:
                 st.error("Failed to connect to OpenAI API after several attempts.")
                 return None
-        except openai.RateLimitError as e:
-            st.error(f"Rate limit reached: {str(e)}")
-            return None
-        except openai.APIError as e:
-            st.error(f"API Error: {str(e)}")
-            return None
-
-# Main logic for the app
+
+# Login page
 if 'username' not in st.session_state:
     # Show the login form if the username is not set
     st.title("Login")
@@ -107,142 +66,108 @@ if 'username' not in st.session_state:
         else:
             st.warning("Please enter a valid username.")
 else:
-    user_id = st.session_state['username']
-    user_data = get_user_data(user_id)
+    # 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:")
     
-    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.")
+    # 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.")
         else:
-            # Main app for generating questions
-            st.title(f"Welcome, {user_id}! Generate your exam questions")
+            # 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.)")
 
-            # 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.")
+    # Select question type
+    question_type = st.selectbox("Select question type:", 
+                                 ["Multiple Choice", "Fill in the Blank", "Open-ended", "True/False"])
 
-            # File uploader for PDFs (limited to 5 MB)
-            uploaded_file = st.file_uploader("Upload a file (PDF)", type="pdf")
+    # 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 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.")
-
-                   
+            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
+            )
+
+            # 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.")