app.py

import gradio as gr
from groq import Groq
import os
from PIL import Image, ImageDraw, ImageFont
from datetime import datetime
import json
import tempfile

# Initialize Groq client
client = Groq(
    api_key=os.getenv("GROQ_API_KEY")
)

class QuizApp:
    def __init__(self):
        self.current_questions = []
        self.user_data = {}
        
    def generate_questions(self, text, num_questions):
        prompt = f"""Generate {num_questions} multiple choice questions based on this text:
        {text}
        
        Format your response as a list of questions, where each question is a JSON object with the following structure:
        [
            {{
                "question": "Question text",
                "options": ["option1", "option2", "option3", "option4"],
                "correct_answers": [0, 1]
            }}
        ]
        Only return the JSON array, no other text."""
        
        try:
            response = client.chat.completions.create(
                messages=[{"role": "user", "content": prompt}],
                model="llama-3.2-3b-preview",
                temperature=0.7,
                max_tokens=1024
            )
            
            # Parse the response directly as a list
            questions = json.loads(response.choices[0].message.content)
            if isinstance(questions, list):
                self.current_questions = questions
            else:
                # If we get an object with a questions key, extract the list
                self.current_questions = questions.get("questions", [])
            
            return json.dumps(self.current_questions, indent=2)
        except Exception as e:
            print(f"Error generating questions: {e}")
            return json.dumps({"error": "Failed to generate valid questions. Please try again."})
    
    def calculate_score(self, answers):
        try:
            answers = json.loads(answers) if isinstance(answers, str) else answers
            total = len(self.current_questions)
            correct = 0
            
            for q, a in zip(self.current_questions, answers):
                if set(a) == set(q["correct_answers"]):
                    correct += 1
                    
            return (correct / total) * 100
        except Exception as e:
            print(f"Error calculating score: {e}")
            return 0
    
    def generate_certificate(self, name, score, course_name, company_logo=None, participant_photo=None):
        try:
            # Create certificate with a light blue background
            certificate = Image.new('RGB', (1200, 800), '#F0F8FF')
            draw = ImageDraw.Draw(certificate)
            
            # Load fonts
            try:
                title_font = ImageFont.truetype("arial.ttf", 60)
                text_font = ImageFont.truetype("arial.ttf", 40)
                subtitle_font = ImageFont.truetype("arial.ttf", 30)
            except:
                print("Using default font as Arial not found")
                title_font = ImageFont.load_default()
                text_font = ImageFont.load_default()
                subtitle_font = ImageFont.load_default()
            
            # Add decorative border
            draw.rectangle([20, 20, 1180, 780], outline='#4682B4', width=3)
            
            # Draw certificate content
            draw.text((600, 100), "CertifyMe AI", font=title_font, fill='#4682B4', anchor="mm")
            draw.text((600, 160), "Certificate of Achievement", font=subtitle_font, fill='#4682B4', anchor="mm")
            draw.text((600, 300), "This is to certify that", font=text_font, fill='black', anchor="mm")
            draw.text((600, 380), name, font=text_font, fill='#4682B4', anchor="mm")
            draw.text((600, 460), "has successfully completed", font=text_font, fill='black', anchor="mm")
            draw.text((600, 540), course_name, font=text_font, fill='#4682B4', anchor="mm")
            draw.text((600, 620), f"with a score of {score:.1f}%", font=text_font, fill='black', anchor="mm")
            
            # Add date
            current_date = datetime.now().strftime("%B %d, %Y")
            draw.text((600, 700), current_date, font=text_font, fill='black', anchor="mm")
            
            # Add logo if provided
            if company_logo is not None:
                try:
                    logo = Image.open(company_logo)
                    logo = logo.resize((150, 150))
                    certificate.paste(logo, (50, 50))
                except Exception as e:
                    print(f"Error adding logo: {e}")
            
            # Add photo if provided
            if participant_photo is not None:
                try:
                    photo = Image.open(participant_photo)
                    photo = photo.resize((150, 150))
                    certificate.paste(photo, (1000, 50))
                except Exception as e:
                    print(f"Error adding photo: {e}")
            
            # Save certificate
            temp_file = tempfile.NamedTemporaryFile(delete=False, suffix='.png')
            certificate.save(temp_file.name)
            return temp_file.name
        except Exception as e:
            print(f"Error generating certificate: {e}")
            return None

def create_quiz_app():
    quiz_app = QuizApp()
    
    with gr.Blocks(title="CertifyMe AI") as demo:
        # Store current tab state
        current_tab = gr.State(0)
        
        gr.Markdown("""
        # CertifyMe AI
        ### Transform Your Knowledge into Recognized Achievements
        Get AI-powered assessments and professional certificates for any learning content.
        """)
        
        # Create tabs
        with gr.Tabs() as tabs:
            # Step 1: Profile Setup
            with gr.Tab("📋 Step 1: Profile Setup", id=0) as tab1:
                with gr.Row():
                    name = gr.Textbox(label="Full Name", placeholder="Enter your full name")
                    email = gr.Textbox(label="Email", placeholder="Enter your email")
                
                text_input = gr.Textbox(
                    label="Learning Content", 
                    placeholder="Enter the text content you want to be assessed on",
                    lines=10
                )
                
                with gr.Row():
                    num_questions = gr.Slider(
                        minimum=1, 
                        maximum=10, 
                        value=5, 
                        step=1,
                        label="Number of Questions"
                    )
                    
                with gr.Row():
                    company_logo = gr.Image(
                        label="Company Logo (Optional)", 
                        type="filepath"
                    )
                    participant_photo = gr.Image(
                        label="Your Photo (Optional)", 
                        type="filepath"
                    )
                    
                generate_btn = gr.Button("Generate Assessment", variant="primary")
                
                # Instructions for Step 1
                gr.Markdown("""
                ### Instructions:
                1. Enter your full name and email
                2. Paste or type the learning content you want to be assessed on
                3. Select the number of questions for your assessment
                4. Optionally upload a company logo and your photo for the certificate
                5. Click 'Generate Assessment' to proceed
                """)
            
            # Step 2: Take Assessment
            with gr.Tab("📝 Step 2: Take Assessment", id=1) as tab2:
                gr.Markdown("""
                ### Assessment Instructions:
                1. Review each question carefully
                2. For each question, you may select one or more correct options
                3. Enter your answers in the following format:
                   ```
                   [
                     [0],      // First question: selected first option
                     [1, 2],   // Second question: selected second and third options
                     [3]       // Third question: selected fourth option
                   ]
                   ```
                4. Click 'Submit Assessment' when you're ready
                5. You need 80% or higher to earn your certificate
                """)
                
                questions_display = gr.JSON(label="Assessment Questions")
                answers_input = gr.JSON(label="Your Answers")
                submit_btn = gr.Button("Submit Assessment", variant="primary")
                score_display = gr.Number(label="Your Score")
            
            # Step 3: Get Certified
            with gr.Tab("🎓 Step 3: Get Certified", id=2) as tab3:
                gr.Markdown("""
                ### Certification
                - Your certificate will be generated automatically if you score 80% or above
                - The certificate includes your name, score, and completion date
                - Company logo and photo will be included if provided
                - You can download your certificate once it's generated
                """)
                
                course_name = gr.Textbox(
                    label="Certification Title", 
                    value="Professional Assessment Certification"
                )
                certificate_display = gr.Image(label="Your Certificate")
        
        # Helper functions for tab navigation
        def generate_and_switch_tab(text, num_questions):
            questions = quiz_app.generate_questions(text, num_questions)
            return questions, 1
            
        def submit_and_switch_tab(answers):
            score = quiz_app.calculate_score(answers)
            return score, 2
        
        # Event handlers
        generate_btn.click(
            fn=generate_and_switch_tab,
            inputs=[text_input, num_questions],
            outputs=[questions_display, current_tab]
        ).then(
            fn=lambda tab: gr.Tabs(selected=tab),
            inputs=[current_tab],
            outputs=[tabs]
        )
        
        submit_btn.click(
            fn=submit_and_switch_tab,
            inputs=[answers_input],
            outputs=[score_display, current_tab]
        ).then(
            fn=lambda tab: gr.Tabs(selected=tab),
            inputs=[current_tab],
            outputs=[tabs]
        )
        
        score_display.change(
            fn=lambda score, user_name, course, logo, photo: (
                quiz_app.generate_certificate(user_name, score, course, logo, photo)
                if score >= 80 else None
            ),
            inputs=[score_display, name, course_name, company_logo, participant_photo],
            outputs=certificate_display
        )
    
    return demo

if __name__ == "__main__":
    if not os.getenv("GROQ_API_KEY"):
        print("Please set your GROQ_API_KEY environment variable")
        exit(1)
        
    demo = create_quiz_app()
    demo.launch()