Update app.py

app.py

@@ -1,11 +1,7 @@
-import os
 import PyPDF2
 import gradio as gr
 from groq import Groq
 
-# Initialize Groq client
-groq_client = Groq(api_key=os.getenv("GROQ_API_KEY"))
-
 # Function to extract PDF 
 def extract_text_from_pdf(pdf_file_path):
     with open(pdf_file_path, 'rb') as file:
@@ -16,7 +12,8 @@ def extract_text_from_pdf(pdf_file_path):
     return extracted_text
 
 # Function to generate different types of questions using Groq
-def generate_questions(paper_text):
+def generate_questions(api_key, paper_text):
+    groq_client = Groq(api_key=api_key)  # Use the provided API key
     prompts = [
         f"Read this paper submitted by a student. Then, ask them to briefly summarize its outline. Ask the question directly. Do not comment on the paper or add anything else. Present this as a reflection exercise. Paper text: '{paper_text}'",
         f"Read this paper submitted by a student. Identify a key term that is central to the discussion. Then, ask the student to explain it in context (provide an excerpt from the paper). Do not comment on the paper or add anything else. Paper text: '{paper_text}'",
@@ -35,7 +32,8 @@ def generate_questions(paper_text):
     return questions
 
 # Function to validate each student's answer 
-def validate_answer(paper_text, question, student_answer):
+def validate_answer(api_key, paper_text, question, student_answer):
+    groq_client = Groq(api_key=api_key)  # Use the provided API key
     prompt = (
         f"The student was asked questions about a paper they submitted. Compare their answers to the content of the paper."
         f"Determine if the student's answer reflects an understanding of the paper's content. The goal is to determine whether there are reasons to believe or to doubt that they understand its content, and therefore that they are its author."
@@ -50,19 +48,19 @@ def validate_answer(paper_text, question, student_answer):
     return response.choices[0].message.content
 
 # Function to handle PDF upload and question generation
-def handle_pdf_upload(pdf_file):
+def handle_pdf_upload(api_key, pdf_file):
     # Save the uploaded file and extract text from it
     pdf_file_path = pdf_file.name
     paper_text = extract_text_from_pdf(pdf_file_path)
     
     # Generate four types of questions based on the paper's content
-    questions = generate_questions(paper_text)
+    questions = generate_questions(api_key, paper_text)
     
     # Return the generated questions for display 
     return questions
 
 # Function to handle answer submission and LLM's final judgment
-def handle_answer_submission(pdf_file, answer1, answer2, answer3, answer4):
+def handle_answer_submission(api_key, pdf_file, answer1, answer2, answer3, answer4):
     # Extract the paper text for validation
     pdf_file_path = pdf_file.name
     paper_text = extract_text_from_pdf(pdf_file_path)
@@ -71,14 +69,14 @@ def handle_answer_submission(pdf_file, answer1, answer2, answer3, answer4):
     answers_summary = ""
     
     # Loop over each question-answer pair and validate
-    questions = generate_questions(paper_text)
+    questions = generate_questions(api_key, paper_text)
     verification_status = "Thank you! "
     emoji = "🟢"
     
     answers = [answer1, answer2, answer3, answer4]
     
     for i in range(len(questions)):
-        validation_result = validate_answer(paper_text, questions[i], answers[i])
+        validation_result = validate_answer(api_key, paper_text, questions[i], answers[i])
         answers_summary += f"Question {i+1}: {questions[i]}\nAnswer: {answers[i]}\nValidation: {validation_result}\n\n"
         if "Flagged" in validation_result:
             verification_status = "Please talk to your teacher about this submission 🟠"
@@ -93,6 +91,9 @@ with gr.Blocks() as app:
         logo_url = "https://i.ibb.co/S7DCk3K/a-logo-design-with-a-stylized-egyptian-scribe-the-i-L59-Gh-Zn-QASp7v-TUazjdp-A-1-s-Ip-Qq-VRIif-VF5dn.png"  # Replace with your actual logo URL
         gr.Image(logo_url, label="", show_label=False, height=200)
 
+    # API key input Section
+    api_key_input = gr.Textbox(label="Enter your Groq API Key", placeholder="You can create a free key at https://console.groq.com/keys")
+
     # PDF upload Section
     pdf_file = gr.File(label="Upload PDF Paper")
     upload_btn = gr.Button("Upload and Generate Questions")
@@ -114,10 +115,10 @@ with gr.Blocks() as app:
     output = gr.Textbox(label="Result", interactive=False)
     
     # Layout: Upload button right under the file upload, followed by Q/A section
-    upload_btn.click(fn=handle_pdf_upload, inputs=pdf_file, outputs=[question1, question2, question3, question4])
+    upload_btn.click(fn=handle_pdf_upload, inputs=[api_key_input, pdf_file], outputs=[question1, question2, question3, question4])
     submit_btn = gr.Button("Submit Answers")
     # Passing answers individually
-    submit_btn.click(fn=handle_answer_submission, inputs=[pdf_file, answer1, answer2, answer3, answer4], outputs=output)
+    submit_btn.click(fn=handle_answer_submission, inputs=[api_key_input, pdf_file, answer1, answer2, answer3, answer4], outputs=output)
     
     # "Powered by Groq" badge 
     with gr.Row():