Update app(backup2).py

app(backup2).py

@@ -6,22 +6,48 @@ import gspread
 from oauth2client.service_account import ServiceAccountCredentials
 import PyPDF2
 import io
+from datetime import datetime
+from PIL import Image
 
-# Set up OpenAI client
-client = OpenAI(api_key=st.secrets["OPENAI_API_KEY"])
+# Add some custom CSS to improve the layout
+st.markdown("""
+    <style>
+    .stImage {
+        text-align: center;
+        display: block;
+        margin-left: auto;
+        margin-right: auto;
+    }
+    
+    .stTitle {
+        text-align: center;
+        padding-bottom: 20px;
+    }
+    
+    div[data-testid="stVerticalBlock"] > div:has(div.stButton) {
+        text-align: center;
+        padding: 10px 0;
+    }
+    </style>
+    """, unsafe_allow_html=True)
 
 # Constants
-WORD_LIMIT = 8000
+WORD_LIMIT = 11000
+DAILY_API_LIMIT = 30  # Set your desired limit per user per day
 
-# Google Sheets setup remains the same
+# Set up OpenAI client
+client = OpenAI(api_key=st.secrets["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
 
 def check_user_in_sheet(username):
+    """Check if user exists in sheet"""
     try:
-        users_list = sheet.col_values(1)
+        users_list = sheet.col_values(1)  # UserID column
         if username in users_list:
             return True
         return False
@@ -29,15 +55,59 @@ def check_user_in_sheet(username):
         st.error(f"Error checking user: {str(e)}")
         return False
 
+def get_user_stats(username):
+    """Get user's current API usage statistics"""
+    try:
+        users_list = sheet.col_values(1)  # UserID column
+        row_number = users_list.index(username) + 1
+        
+        daily_count = int(sheet.cell(row_number, 2).value)    # DailyAPICount
+        total_count = int(sheet.cell(row_number, 3).value)    # TotalAPICount
+        last_used = sheet.cell(row_number, 4).value           # LastUsedDate
+        
+        return {
+            'daily_count': daily_count,
+            'total_count': total_count,
+            'last_used': last_used
+        }
+    except Exception as e:
+        st.error(f"Error getting user stats: {str(e)}")
+        return None
+
 def update_api_usage(username):
+    """Update both daily and total API usage counts"""
     try:
         users_list = sheet.col_values(1)
         row_number = users_list.index(username) + 1
-        api_usage = int(sheet.cell(row_number, 2).value)
-        api_usage += 1
-        sheet.update_cell(row_number, 2, api_usage)
+        today = datetime.now().strftime('%Y-%m-%d')
+        
+        # Get current values
+        stats = get_user_stats(username)
+        if not stats:
+            return False, "Error retrieving user statistics"
+        
+        # Reset daily count if it's a new day
+        daily_count = stats['daily_count']
+        if stats['last_used'] != today:
+            daily_count = 0
+        
+        # Check daily limit
+        if daily_count >= DAILY_API_LIMIT:
+            return False, f"You have reached your daily limit of {DAILY_API_LIMIT} generations. Please try again tomorrow."
+        
+        # Update counts
+        new_daily_count = daily_count + 1
+        new_total_count = stats['total_count'] + 1
+        
+        # Update all values in sheet
+        sheet.update_cell(row_number, 2, new_daily_count)  # Update DailyAPICount
+        sheet.update_cell(row_number, 3, new_total_count)  # Update TotalAPICount
+        sheet.update_cell(row_number, 4, today)            # Update LastUsedDate
+        
+        return True, None
+        
     except Exception as e:
-        st.error(f"Error updating API usage: {str(e)}")
+        return False, f"Error updating API usage: {str(e)}"
 
 def extract_text_from_pdf(pdf_file):
     """Simple PDF text extraction with word limit check"""
@@ -54,7 +124,15 @@ def extract_text_from_pdf(pdf_file):
     except Exception as e:
         return None, f"Error processing PDF: {str(e)}"
 
-def generate_questions_with_retry(knowledge_material, question_type, cognitive_level, extra_instructions, case_based, num_choices=None, max_retries=3):
+def generate_questions_with_retry(username, knowledge_material, question_type, cognitive_level, extra_instructions, case_based, num_choices=None, max_retries=3):
+    """Generate questions and update API usage"""
+    
+    # Check and update API usage before generating
+    can_generate, error_message = update_api_usage(username)
+    if not can_generate:
+        st.error(error_message)
+        return None
+    
     # Adjust number of questions based on type
     if question_type == "Multiple Choice":
         num_questions = 3
@@ -65,7 +143,6 @@ def generate_questions_with_retry(knowledge_material, question_type, cognitive_l
         3. After all questions, provide an ANSWER KEY section with:
            - The correct answer letter for each question
            - A brief explanation of why this is the correct answer
-           - Why other options are incorrect
         """
     elif question_type == "Fill in the Blank":
         num_questions = 10
@@ -162,23 +239,68 @@ Please format the output with:
                 return None
             time.sleep(2)
 
-# ระบบ login
 # Main Streamlit interface
+# Initialize session state variables
+if 'login_step' not in st.session_state:
+    st.session_state.login_step = 'username'
 if 'username' not in st.session_state:
-    st.title("Login")
-    username_input = st.text_input("Enter your username:")
-    if st.button("Login"):
-        if username_input:
-            if check_user_in_sheet(username_input):
-                st.session_state['username'] = username_input
-                st.success(f"Welcome, {username_input}!")
-                update_api_usage(username_input)
+    st.session_state.username = None
+
+# Login system
+if st.session_state.username is None:
+    # Center align the content
+    col1, col2, col3 = st.columns([1,2,1])
+    
+    with col2:
+        # Display logo
+        st.image("GenExam.png", width=200)  # Assuming the image is saved as logo.png
+        
+        st.title("Login")
+        username_input = st.text_input("Enter your username:")
+
+        if st.session_state.login_step == 'username' and st.button("Login", use_container_width=True):
+            if username_input:
+                if check_user_in_sheet(username_input):
+                    stats = get_user_stats(username_input)
+                    if stats:
+                        st.success(f"Welcome, {username_input}! 👋")
+                        st.info(f"""
+                        📊 Your API Usage Statistics:
+                        - Today's Usage: {stats['daily_count']}/{DAILY_API_LIMIT} generations
+                        - Total All-Time Usage: {stats['total_count']} generations
+                        """)
+                        st.session_state.login_step = 'enter_app'
+                else:
+                    st.warning("Username not found. Please try again.")
             else:
-                st.warning("Username not found. Please try again.")
-        else:
-            st.warning("Please enter a valid username.")
+                st.warning("Please enter a valid username.")
+
+        if st.session_state.login_step == 'enter_app':
+            if st.button("🎯 Enter GeneXam Application", use_container_width=True):
+                st.session_state.username = username_input
+                st.rerun()
+
+        # Show instructions
+        if st.session_state.login_step == 'username':
+            st.markdown("""
+            ### How to Login:
+            1. Enter your username and click 'Login' to verify your account
+            2. After verification, click 'Enter GeneXam Application' to start using the system
+            """)
+
 else:
-    st.title(f"Welcome, {st.session_state['username']}! Generate your exam questions")
+    # Main application code (ส่วนที่เหลือเหมือนเดิม)
+    st.title(f"Welcome to GeneXam, {st.session_state.username}! 🎓")
+    
+    # Show current usage stats
+    stats = get_user_stats(st.session_state.username)
+    if stats:
+        remaining = DAILY_API_LIMIT - stats['daily_count']
+        st.info(f"""
+        📊 Usage Statistics:
+        - Daily Generations Remaining: {remaining}/{DAILY_API_LIMIT}
+        - Total All-Time Generations: {stats['total_count']}
+        """)
     
     # Create tabs for input methods
     tab1, tab2 = st.tabs(["Text Input", "PDF Upload"])
@@ -226,8 +348,8 @@ else:
     if st.button("Generate Questions"):
         if 'knowledge_material' in locals() and knowledge_material.strip():
             with st.spinner("Generating questions..."):
-                # Your existing generate_questions_with_retry function call here
                 questions = generate_questions_with_retry(
+                    st.session_state['username'],
                     knowledge_material,
                     question_type,
                     cognitive_level,
@@ -240,6 +362,16 @@ else:
                     st.write("### Generated Exam Questions:")
                     st.write(questions)
                     
+                    # Update displayed stats after generation
+                    new_stats = get_user_stats(st.session_state['username'])
+                    if new_stats:
+                        remaining = DAILY_API_LIMIT - new_stats['daily_count']
+                        st.info(f"""
+                        📊 Updated Usage Statistics:
+                        - Daily Generations Remaining: {remaining}/{DAILY_API_LIMIT}
+                        - Total All-Time Generations: {new_stats['total_count']}
+                        """)
+                    
                     # Download button
                     st.download_button(
                         label="Download Questions",