Update app.py

app.py

@@ -1,12 +1,14 @@
 import os
 import re
-from datetime import datetime
 
 import gradio as gr
-import openai
+from openai import OpenAI
 
-# Set OpenAI API key from an environment variable.
-openai.api_key = os.environ["OPENAI_API_KEY"]
+# Initialize the OpenAI client with the API key from environment variables.
+client = OpenAI(api_key=os.environ["OPENAI_API_KEY"])
+
+# In-memory storage to track submitted emails (not persistent; resets on app restart).
+submitted_emails = set()
 
 def get_evaluation_questions():
     """
@@ -37,77 +39,107 @@ def sanitize_input(text):
     clean_text = re.sub(r"[^a-zA-Z0-9\s.,!?@:\-]", "", text)
     return clean_text.strip()[:500]
 
-def evaluate_prompt(email, name, system_prompt):
+def validate_email(email):
+    """
+    Validates that the provided email is in a valid format.
+    Returns True if valid, False otherwise.
     """
-    For each test question:
-      - Uses the provided system prompt to generate a response with GPT-4o Mini.
-      - Checks if the expected substring is present.
-      - Computes an aggregate score.
-    Returns the evaluation results as a string.
+    email_regex = r"^[a-zA-Z0-9_.+-]+@[a-zA-Z0-9-]+\.[a-zA-Z0-9-.]+$"
+    return re.match(email_regex, email) is not None
+
+def submit_prompt(email, name, system_prompt):
     """
-    # Sanitize the inputs.
+    Handles user submission:
+      - Validates email format.
+      - Checks if the email has already been used for submission.
+      - Evaluates the system prompt against predefined test questions.
+      - Prevents multiple submissions from the same email.
+    Returns the evaluation results or an error message if the submission is invalid.
+    """
+    # Validate email format.
+    if not validate_email(email):
+        return "Invalid email address. Please enter a valid email."
+
+    # Check if this email has already been used for submission.
+    if email in submitted_emails:
+        return f"Submission already received for {email}. You can only submit once."
+
+    # Sanitize inputs.
     email = sanitize_input(email)
     name = sanitize_input(name)
     system_prompt = sanitize_input(system_prompt)
 
     score = 0
     responses = []
+
     for item in EVALUATION_QUESTIONS:
         question = item["question"]
         expected = item["expected"]
         try:
-            response = openai.ChatCompletion.create(
+            # Use the new client-based API for chat completions.
+            response = client.chat.completions.create(
                 model="gpt-4o-mini",  # Ensure this identifier matches the deployed model.
                 messages=[
                     {"role": "system", "content": system_prompt},
                     {"role": "user", "content": question}
                 ]
             )
-            answer = response.choices[0].message["content"].strip()
+            # Extract the answer from the response object.
+            answer = response.choices[0].message.content.strip()
         except Exception as e:
             answer = f"Error during OpenAI API call: {str(e)}"
-        
+
         # Simple evaluation: check if the answer contains the expected substring.
         if expected.lower() in answer.lower():
             score += 1
             verdict = "Correct"
         else:
             verdict = "Incorrect"
-        
+
         responses.append(
             f"Question: {question}\n"
             f"Answer: {answer}\n"
             f"Expected: {expected}\n"
             f"Result: {verdict}\n"
         )
-    
+
     result_details = "\n".join(responses)
-    
-    return f"Your evaluation score is {score} out of {len(EVALUATION_QUESTIONS)}.\n\nDetails:\n{result_details}"
+
+    # Record this email as having submitted their prompt.
+    submitted_emails.add(email)
+
+    return (
+        f"Thank you for your submission, {name}!\n\n"
+        f"Your evaluation score is {score} out of {len(EVALUATION_QUESTIONS)}.\n\nDetails:\n{result_details}"
+    )
 
 def build_interface():
     """
-    Constructs the Gradio interface.
+    Constructs the Gradio interface with a submission button and single-submission mechanism.
     """
     with gr.Blocks() as demo:
-        gr.Markdown("# GPT-4o Mini Prompt Evaluation")
-        gr.Markdown("Enter your email, name, and a system prompt below:")
-        
+        gr.Markdown("# GPT-4o Mini Prompt Submission")
+        gr.Markdown(
+            "Please enter your details and submit your system prompt below. "
+            "You can only submit once."
+        )
+
         email_input = gr.Textbox(label="Email", placeholder="[email protected]")
         name_input = gr.Textbox(label="Name", placeholder="Your name")
         system_prompt_input = gr.Textbox(
             label="System Prompt",
             placeholder="Enter your system prompt here...",
-            lines=6
+            lines=6,
         )
-        eval_button = gr.Button("Evaluate")
+        submit_button = gr.Button("Submit")
         output_text = gr.Textbox(label="Results", lines=15)
-        
-        eval_button.click(
-            fn=evaluate_prompt,
+
+        submit_button.click(
+            fn=submit_prompt,
             inputs=[email_input, name_input, system_prompt_input],
-            outputs=output_text
+            outputs=output_text,
         )
+    
     return demo
 
 if __name__ == "__main__":