added topics dictionary, added randomized difficulty distribution, made temperature dynamic

app.py

@@ -1,17 +1,17 @@
-# app.py
 import os
 import gradio as gr
 from anthropic import Anthropic
 from datetime import datetime, timedelta
 from collections import deque
+import random
 
-# Initialize Anthropic client - will use the secret key from HuggingFace
+# Initialize Anthropic client
 anthropic = Anthropic(
     api_key=os.environ.get('ANTHROPIC_API_KEY')
 )
 
 # Request tracking
-MAX_REQUESTS_PER_DAY = 25  # Conservative limit to start
+MAX_REQUESTS_PER_DAY = 25
 request_history = deque(maxlen=1000)
 
 def check_api_key():
@@ -22,7 +22,6 @@ def check_api_key():
 def check_rate_limit():
     """Check if we're within rate limits"""
     now = datetime.now()
-    # Remove requests older than 24 hours
     while request_history and (now - request_history[0]) > timedelta(days=1):
         request_history.popleft()
     return len(request_history) < MAX_REQUESTS_PER_DAY
@@ -33,33 +32,55 @@ def clean_latex(text):
     return text
 
 def generate_test(subject):
-    """Generate a math test with error handling and rate limiting"""
+    """Generate a math test with enhanced variety"""
     try:
-        # Check API key
         check_api_key()
-        
-        # Check rate limit
         if not check_rate_limit():
             return "Daily request limit reached. Please try again tomorrow."
         
-        # Record request
         request_history.append(datetime.now())
         
-        system_prompt = """You will write math exam questions. Follow these requirements EXACTLY:
-        1. Write exactly 3 challenging university-level questions
-        2. For LaTeX math formatting:
+        # Randomly select focus areas and difficulty levels
+        topics = {
+            "Single Variable Calculus": ["limits", "derivatives", "integrals", "series", "applications"],
+            "Multivariable Calculus": ["partial derivatives", "multiple integrals", "vector fields", "optimization", "surface integrals"],
+            "Linear Algebra": ["matrices", "vector spaces", "eigenvalues", "linear transformations", "inner products"],
+            # ... add topics for other subjects
+        }
+        
+        selected_topics = random.sample(topics.get(subject, ["general"]), min(3, len(topics.get(subject, ["general"]))))
+        difficulty_distribution = random.choice([
+            "one easy, one medium, one challenging",
+            "two medium, one very challenging",
+            "one medium, two challenging"
+        ])
+        
+        system_prompt = f"""You will write math exam questions. Follow these requirements EXACTLY:
+        1. Write exactly 3 university-level questions focusing on these specific topics: {', '.join(selected_topics)}
+        2. Make the questions {difficulty_distribution}
+        3. For each question, use a different type of mathematical thinking (e.g., computation, proof, application, conceptual understanding)
+        4. Ensure questions are unique and not commonly found in textbooks
+        5. For LaTeX math formatting:
            - Use $ for simple inline math
            - For equations and solution steps, use $$ on separate lines
            - For multi-step solutions, put each step on its own line in $$ $$
-           - DO NOT use \\begin{aligned} or any other environments
-        3. Number each question as 1), 2), 3)
-        4. Include solutions after each question
-        5. Keep formatting simple and clear"""
+           - DO NOT use \\begin{{aligned}} or any other environments
+        6. Number each question as 1), 2), 3)
+        7. Include detailed solutions after each question
+        8. Keep formatting simple and clear
+        
+        Important: Make these questions distinct from standard textbook problems. Incorporate real-world applications, 
+        interdisciplinary connections, or creative scenarios where appropriate."""
+        
+        # Vary temperature based on subject complexity
+        base_temperature = 0.7
+        complexity_adjustment = random.uniform(-0.1, 0.1)
+        temperature = min(1.0, max(0.5, base_temperature + complexity_adjustment))
         
         message = anthropic.messages.create(
             model="claude-3-opus-20240229",
             max_tokens=1500,
-            temperature=0.7,
+            temperature=temperature,
             messages=[{
                 "role": "user",
                 "content": f"{system_prompt}\n\nWrite an exam for {subject}."
@@ -97,7 +118,7 @@ def generate_test(subject):
     except Exception as e:
         return f"Error: {str(e)}"
 
-# Subject choices
+# Subject choices remain the same
 subjects = [
     "Single Variable Calculus",
     "Multivariable Calculus", 
@@ -127,12 +148,11 @@ interface = gr.Interface(
         ]
     ),
     title="Advanced Mathematics Test Generator",
-    description="""Generates university-level mathematics exam questions with solutions using Claude 3 Opus.
-    Limited to 25 requests per day. Please use responsibly.""",
+    description="""Generates unique university-level mathematics exam questions with solutions using Claude 3 Opus.
+    Each test features different topics and difficulty levels. Limited to 25 requests per day.""",
     theme="default",
     allow_flagging="never"
 )
 
-# Launch the interface
 if __name__ == "__main__":
-    interface.launch()
+    interface.launch()