Update app.py

app.py

@@ -4,9 +4,6 @@ from anthropic import Anthropic
 from datetime import datetime, timedelta
 from collections import deque
 import random
-import tempfile
-import subprocess
-from pathlib import Path
 
 # Initialize Anthropic client
 anthropic = Anthropic(
@@ -37,47 +34,41 @@ def clean_latex(text):
 def create_latex_document(content, questions_only=False):
     """Create a complete LaTeX document"""
     latex_header = r"""\documentclass{article}
-    \usepackage{amsmath,amssymb}
-    \usepackage[margin=1in]{geometry}
-    \begin{document}
-    """
+\usepackage{amsmath,amssymb}
+\usepackage[margin=1in]{geometry}
+\begin{document}
+\title{Mathematics Test}
+\maketitle
+"""
     latex_footer = r"\end{document}"
     
     if questions_only:
         # Extract only the questions (everything before "Solution:")
-        questions = []
-        for question in content.split('\n'):
-            if 'Solution:' in question:
+        processed_content = []
+        current_question = []
+        for line in content.split('\n'):
+            if 'Solution:' in line:
+                processed_content.append('\n'.join(current_question))
+                current_question = []
                 continue
-            questions.append(question)
-        content = '\n'.join(questions)
+            if any(line.startswith(f"{i})") for i in range(1, 4)):
+                if current_question:
+                    processed_content.append('\n'.join(current_question))
+                current_question = [line]
+            elif current_question:
+                current_question.append(line)
+        if current_question:
+            processed_content.append('\n'.join(current_question))
+        content = '\n\n'.join(processed_content)
     
     return f"{latex_header}\n{content}\n{latex_footer}"
 
-def latex_to_pdf(latex_content):
-    """Convert LaTeX content to PDF"""
-    with tempfile.TemporaryDirectory() as tmpdir:
-        tex_path = Path(tmpdir) / "output.tex"
-        with open(tex_path, "w", encoding="utf-8") as f:
-            f.write(latex_content)
-        
-        # Run pdflatex twice to resolve references
-        subprocess.run(["pdflatex", "-interaction=nonstopmode", str(tex_path)], 
-                     cwd=tmpdir, capture_output=True)
-        subprocess.run(["pdflatex", "-interaction=nonstopmode", str(tex_path)], 
-                     cwd=tmpdir, capture_output=True)
-        
-        pdf_path = Path(tmpdir) / "output.pdf"
-        if pdf_path.exists():
-            return pdf_path.read_bytes()
-        return None
-
-def generate_test_with_format(subject, output_format):
-    """Generate a math test with specified output format"""
+def generate_test(subject):
+    """Generate a math test"""
     try:
         check_api_key()
         if not check_rate_limit():
-            return "Daily request limit reached. Please try again tomorrow."
+            return "Daily request limit reached. Please try again tomorrow.", None, None
         
         request_history.append(datetime.now())
         
@@ -111,12 +102,10 @@ def generate_test_with_format(subject, output_format):
         
         Important: Every question must require mathematical computation or proof. Do not ask for explanations, descriptions, or concepts."""
         
-        temperature = 0.7
-        
         message = anthropic.messages.create(
             model="claude-3-opus-20240229",
             max_tokens=1500,
-            temperature=temperature,
+            temperature=0.7,
             messages=[{
                 "role": "user",
                 "content": f"{system_prompt}\n\nWrite an exam for {subject}."
@@ -125,29 +114,18 @@ def generate_test_with_format(subject, output_format):
         
         if hasattr(message, 'content') and len(message.content) > 0:
             response_text = message.content[0].text
-            return clean_latex(response_text)
+            cleaned_text = clean_latex(response_text)
+            
+            # Create both versions of the LaTeX document
+            full_latex = create_latex_document(cleaned_text, questions_only=False)
+            questions_latex = create_latex_document(cleaned_text, questions_only=True)
+            
+            return cleaned_text, questions_latex, full_latex
         else:
-            return "Error: No content in response"
+            return "Error: No content in response", None, None
             
     except Exception as e:
-        return f"Error: {str(e)}"
-
-def create_download_files(test_content):
-    """Create downloadable files in different formats"""
-    # Full test (questions and solutions)
-    full_latex = create_latex_document(test_content, questions_only=False)
-    full_pdf = latex_to_pdf(full_latex)
-    
-    # Questions only
-    questions_latex = create_latex_document(test_content, questions_only=True)
-    questions_pdf = latex_to_pdf(questions_latex)
-    
-    return {
-        "full_latex": full_latex,
-        "full_pdf": full_pdf,
-        "questions_latex": questions_latex,
-        "questions_pdf": questions_pdf
-    }
+        return f"Error: {str(e)}", None, None
 
 subjects = [
     "Single Variable Calculus",
@@ -162,10 +140,6 @@ subjects = [
     "Topology"
 ]
 
-def download_handler(file_content, filename):
-    """Handle file downloads"""
-    return file_content
-
 # Create Gradio interface
 with gr.Blocks() as interface:
     gr.Markdown("# Advanced Mathematics Test Generator")
@@ -190,45 +164,24 @@ with gr.Blocks() as interface:
     
     with gr.Row():
         with gr.Column():
-            gr.Markdown("### Download Questions Only")
-            questions_latex_btn = gr.Button("Download Questions (LaTeX)")
-            questions_pdf_btn = gr.Button("Download Questions (PDF)")
-        
-        with gr.Column():
-            gr.Markdown("### Download Full Test with Solutions")
-            full_latex_btn = gr.Button("Download Full Test (LaTeX)")
-            full_pdf_btn = gr.Button("Download Full Test (PDF)")
-    
-    # Hidden components for downloads
-    questions_latex_output = gr.File(label="Questions LaTeX", visible=False)
-    questions_pdf_output = gr.File(label="Questions PDF", visible=False)
-    full_latex_output = gr.File(label="Full LaTeX", visible=False)
-    full_pdf_output = gr.File(label="Full PDF", visible=False)
+            gr.Markdown("### Download LaTeX Files")
+            questions_file = gr.File(label="Questions Only (LaTeX)")
+            full_file = gr.File(label="Full Test with Solutions (LaTeX)")
     
-    # Event handlers
-    def generate_and_prepare_downloads(subject):
-        test_content = generate_test_with_format(subject, "latex")
-        files = create_download_files(test_content)
-        return (
-            test_content,
-            (files["questions_latex"], "questions.tex"),
-            (files["questions_pdf"], "questions.pdf"),
-            (files["full_latex"], "full_test.tex"),
-            (files["full_pdf"], "full_test.pdf")
-        )
+    def generate_and_prepare_files(subject):
+        preview, questions_latex, full_latex = generate_test(subject)
+        
+        # Create temporary files for download
+        questions_file = {"content": questions_latex.encode(), "name": "questions.tex"}
+        full_file = {"content": full_latex.encode(), "name": "full_test.tex"}
+        
+        return preview, questions_file, full_file
     
     generate_btn.click(
-        generate_and_prepare_downloads,
+        generate_and_prepare_files,
         inputs=[subject_dropdown],
-        outputs=[output_text, questions_latex_output, questions_pdf_output, 
-                full_latex_output, full_pdf_output]
+        outputs=[output_text, questions_file, full_file]
     )
-    
-    # Download button clicks
-    questions_latex_btn.click(lambda: questions_latex_output.value)
-    questions_pdf_btn.click(lambda: questions_pdf_output.value)
-    full_latex_btn.click(lambda: full_latex_output.value)
-    full_pdf_btn.click(lambda: full_pdf_output.value)
 
 if __name__ == "__main__":
     interface.launch()