Update app.py

app.py

@@ -1,14 +1,53 @@
 import streamlit as st
 import openai
 import time
-
-# Set your OpenAI API key from Hugging Face Secrets
+import gspread
+from oauth2client.service_account import ServiceAccountCredentials
+
+# ตั้งค่า Google Sheets API
+scope = ["https://spreadsheets.google.com/feeds", "https://www.googleapis.com/auth/drive"]
+creds = ServiceAccountCredentials.from_json_keyfile_name("genexam-2c8c645ecc0d.json", scope)  # เปลี่ยนเป็นชื่อไฟล์ JSON ของคุณ
+client_gs = gspread.authorize(creds)
+
+# เปิด Google Sheets โดยใช้ชื่อของ sheet
+sheet = client_gs.open("GeneXam user").sheet1  # ใส่ชื่อจริงของ Google Sheet
+
+# ฟังก์ชันตรวจสอบ username ใน Google Sheets
+def check_user_in_sheet(username):
+    try:
+        users_list = sheet.col_values(1)  # สมมติว่า usernames อยู่ในคอลัมน์แรก
+        if username in users_list:
+            return True
+        else:
+            return False
+    except Exception as e:
+        st.error(f"Error checking user: {str(e)}")
+        return False
+
+# ฟังก์ชันอัปเดตการใช้ API ของผู้ใช้
+def update_api_usage(username):
+    try:
+        users_list = sheet.col_values(1)
+        row_number = users_list.index(username) + 1  # หาตำแหน่งของ user
+        api_usage = int(sheet.cell(row_number, 2).value)  # คอลัมน์ที่ 2 เป็นจำนวนการใช้ API
+        api_usage += 1
+        sheet.update_cell(row_number, 2, api_usage)
+    except Exception as e:
+        st.error(f"Error updating API usage: {str(e)}")
+
+# รีเซ็ตการใช้ API เมื่อถึงเวลารีเซ็ตทุกวัน (ถ้าจำเป็น)
+def reset_api_usage_daily():
+    try:
+        users_list = sheet.col_values(1)
+        for i, user in enumerate(users_list):
+            sheet.update_cell(i + 1, 2, 0)  # รีเซ็ตคอลัมน์ที่ 2 (การใช้ API) เป็น 0
+    except Exception as e:
+        st.error(f"Error resetting API usage: {str(e)}")
+
+# ตั้งค่า OpenAI API key จาก Hugging Face Secrets
 openai.api_key = st.secrets["OPENAI_API_KEY"]
 
-# Initialize OpenAI client
-client = openai.OpenAI(api_key=openai.api_key)
-
-# Function to generate exam questions using OpenAI API with retry logic
+# ฟังก์ชันการสร้างข้อสอบโดยใช้ OpenAI API และมี 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
     if question_type == "Multiple Choice":
@@ -39,14 +78,16 @@ def generate_questions_with_retry(knowledge_material, question_type, cognitive_l
     retries = 0
     while retries < max_retries:
         try:
-            response = client.chat.completions.create(
-                model="gpt-4o-mini",
+            response = openai.ChatCompletion.create(
+                model="gpt-4o-mini",  # You can switch to gpt-4 if available in your plan
                 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
+            return response.choices[0].message["content"]
         except openai.error.APIConnectionError:
             retries += 1
             time.sleep(2)  # Wait for 2 seconds before retrying
@@ -54,15 +95,18 @@ 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
+# ระบบ login
 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}!")
+            if check_user_in_sheet(username_input):
+                st.session_state['username'] = username_input
+                st.success(f"Welcome, {username_input}!")
+                update_api_usage(username_input)
+            else:
+                st.warning("Username not found. Please try again.")
         else:
             st.warning("Please enter a valid username.")
 else:
@@ -83,8 +127,6 @@ else:
         if uploaded_file.size > 5 * 1024 * 1024:  # 5 MB limit
             st.warning("File size exceeds 5 MB. Please upload a smaller file.")
         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