Update app.py

app.py

@@ -9,11 +9,12 @@ from groq import Groq
 groq_client = Groq(api_key=os.getenv("GROQ_API_KEY"))
 
 # PDF Extraction Function
-def extract_text_from_pdf(pdf_file):
-    reader = PyPDF2.PdfReader(pdf_file)
-    extracted_text = ""
-    for page in reader.pages:
-        extracted_text += page.extract_text()
+def extract_text_from_pdf(pdf_file_path):
+    with open(pdf_file_path, 'rb') as file:
+        reader = PyPDF2.PdfReader(file)
+        extracted_text = ""
+        for page in reader.pages:
+            extracted_text += page.extract_text()
     return extracted_text
 
 # Function to generate a question using Groq (Improved Prompt)
@@ -30,7 +31,7 @@ def generate_question(paper_text):
         messages=[{"role": "user", "content": prompt}],
         model="llama-3.1-70b-versatile"
     )
-    # Access the message content properly, no subscripting
+    # Access the message content properly
     return response.choices[0].message.content
 
 # Validate the student's answer using Groq (Improved Prompt)
@@ -50,42 +51,23 @@ def validate_answer(paper_text, student_answer):
         messages=[{"role": "user", "content": prompt}],
         model="llama-3.1-70b-versatile"
     )
-    # Access the message content properly, no subscripting
+    # Access the message content properly
     return response.choices[0].message.content
 
-# Function to determine authorship based on multiple interactions
-def determine_authorship(paper_text, student_answers, interaction_count):
-    # Accumulate evidence based on the student's answers over multiple rounds
-    prompt = (
-        f"Based on the student’s answers to several questions, evaluate whether the student is likely the author of the paper. "
-        f"Consider the depth of understanding reflected in their responses, their ability to explain concepts in the paper, and "
-        f"whether their explanations match the complexity of the original text. If their answers show gaps in understanding or seem "
-        f"inconsistent with the paper's content, note these concerns. If after {interaction_count} interactions, the student consistently "
-        f"shows a deep understanding, confirm likely authorship. Otherwise, suggest that authorship is uncertain.\n\n"
-        f"Paper text: '{paper_text}'\n"
-        f"Student’s responses: '{student_answers}'"
-    )
-    
-    response = groq_client.chat.completions.create(
-        messages=[{"role": "user", "content": prompt}],
-        model="llama-3.1-70b-versatile"
-    )
-    # Access the message content properly, no subscripting
-    return response.choices[0].message.content
-
-# Mailjet Email Sending Function
-def send_email(pdf_file, student_answer, verification_status):
+# Mailjet Email Sending Function (Corrected)
+def send_email(pdf_file_path, student_answer, verification_status):
     api_key = os.getenv("MAILJET_API_KEY")
     api_secret = os.getenv("MAILJET_SECRET_KEY")
     mailjet = Client(auth=(api_key, api_secret), version='v3.1')
     
+    # Open and read the PDF file as binary content
+    with open(pdf_file_path, "rb") as file:
+        attachment_content = base64.b64encode(file.read()).decode('utf-8')
+    
     # Determine the email subject and content based on verification status
     subject = "Authorship Confirmation" if verification_status == "Confirmed" else "Authorship Concerns"
     message_content = f"Authorship Status: {verification_status}\n\nStudent Answer: {student_answer}"
     
-    # Attach the original PDF in the email
-    attachment_content = base64.b64encode(pdf_file.read()).decode('utf-8')
-    
     data = {
       'Messages': [
         {
@@ -104,7 +86,7 @@ def send_email(pdf_file, student_answer, verification_status):
           "Attachments": [
             {
               "ContentType": "application/pdf",
-              "Filename": pdf_file.name,
+              "Filename": os.path.basename(pdf_file_path),
               "Base64Content": attachment_content
             }
           ]
@@ -115,38 +97,50 @@ def send_email(pdf_file, student_answer, verification_status):
     return result.status_code
 
 # Gradio Interface for the main app
-def main_app(pdf_file, student_answer, interaction_count=3):
-    # Extract text from the PDF
-    paper_text = extract_text_from_pdf(pdf_file)
+def handle_pdf_upload(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 a list of questions to challenge the student's understanding
-    student_answers = []
-    for i in range(interaction_count):
-        question = generate_question(paper_text)
-        student_answers.append(f"Q{i+1}: {question}\nA{i+1}: {student_answer}")
+    # Generate the first question based on the paper's content
+    question = generate_question(paper_text)
     
-    # Validate the student's answer for each question
-    final_verification = determine_authorship(paper_text, student_answers, interaction_count)
+    # Return the generated question for display to the user
+    return question
+
+def handle_answer_submission(pdf_file, student_answer):
+    # Extract the paper text for validation
+    pdf_file_path = pdf_file.name
+    paper_text = extract_text_from_pdf(pdf_file_path)
+    
+    # Validate the student's answer
+    validation_result = validate_answer(paper_text, student_answer)
     
-    # Send an email based on the final verification status
-    verification_status = "Confirmed" if "likely the author" in final_verification else "Not Confirmed"
-    send_email(pdf_file, student_answer, verification_status)
+    # Determine verification status based on the validation result
+    verification_status = "Confirmed" if "likely understanding" in validation_result else "Not Confirmed"
     
-    return f"Final Verification: {final_verification}\nStatus: {verification_status}"
+    # Send an email to the teacher with the result and the PDF attachment
+    send_email(pdf_file_path, student_answer, verification_status)
+    
+    return f"Final Verification: {validation_result}\nStatus: {verification_status}"
 
 # Gradio Blocks for the App
 with gr.Blocks() as app:
     gr.Markdown("# Authorship Verification App")
     
     pdf_file = gr.File(label="Upload PDF Paper")
+    question_display = gr.Textbox(label="Generated Question", interactive=False)
     student_answer = gr.Textbox(label="Student Answer", placeholder="Answer the question here...")
     
+    upload_btn = gr.Button("Upload and Generate Question")
     submit_btn = gr.Button("Submit Answer")
-    output = gr.Textbox(label="Result")
+    output = gr.Textbox(label="Result", interactive=False)
+    
+    # Upon PDF upload, generate and display the question
+    upload_btn.click(fn=handle_pdf_upload, inputs=pdf_file, outputs=question_display)
     
-    # Ensure that the file upload triggers the processing and the student can submit their answer
-    pdf_file.upload(fn=extract_text_from_pdf, inputs=pdf_file, outputs=None)
-    submit_btn.click(fn=main_app, inputs=[pdf_file, student_answer], outputs=output)
+    # Upon answer submission, validate the answer and return the result
+    submit_btn.click(fn=handle_answer_submission, inputs=[pdf_file, student_answer], outputs=output)
 
 # Launch the app
 app.launch()