Create app.py

app.py

@@ -0,0 +1,359 @@
+import gradio as gr
+from groq import Groq
+from fpdf import FPDF
+import json
+import os
+import re
+from datetime import datetime
+
+# Initialize Groq client from environment variable
+GROQ_API_KEY = os.environ.get("GROQ_API_KEY")
+if not GROQ_API_KEY:
+    raise ValueError("GROQ_API_KEY not found in environment variables. Please set it in Space Secrets.")
+
+client = Groq(api_key=GROQ_API_KEY)
+
+def clean_json_string(content):
+    """Remove invalid control characters from JSON string"""
+    content = re.sub(r'[\x00-\x08\x0B\x0C\x0E-\x1F\x7F-\x9F]', '', content)
+    return content
+
+def generate_roadmap(domain, level, time_available, time_unit):
+    """Generate roadmap using Groq API"""
+    if time_unit == "Weeks":
+        total_weeks = int(time_available)
+    elif time_unit == "Months":
+        total_weeks = int(time_available) * 4
+    elif time_unit == "Years":
+        total_weeks = int(time_available) * 52
+    
+    system_prompt = """You are an expert educational curriculum designer. Return only valid JSON without any markdown formatting or explanatory text."""
+    
+    user_prompt = f"""Create a detailed learning roadmap for {domain} at {level} level.
+Duration: {time_available} {time_unit.lower()} ({total_weeks} weeks).
+
+STRICT REQUIREMENTS:
+1. Return ONLY valid JSON, no markdown, no backticks, no explanation
+2. Use this exact structure:
+{{
+    "domain": "{domain}",
+    "level": "{level}",
+    "duration": "{time_available} {time_unit}",
+    "overview": "Brief overview text here",
+    "phases": [
+        {{
+            "phase_name": "Phase Name",
+            "duration_weeks": number,
+            "description": "Description here",
+            "topics": [
+                {{
+                    "name": "Topic name",
+                    "resources": ["resource 1", "resource 2"],
+                    "practice_project": "Project description"
+                }}
+            ],
+            "milestones": ["milestone 1", "milestone 2"]
+        }}
+    ],
+    "essential_resources": {{
+        "courses": ["course 1"],
+        "books": ["book 1"],
+        "communities": ["community 1"],
+        "tools": ["tool 1"]
+    }},
+    "tips_for_success": ["tip 1", "tip 2"],
+    "next_steps": "What to do next"
+}}
+
+3. Ensure all strings are properly escaped (no raw newlines or tabs in strings)
+4. Use \\n for newlines within strings if needed"""
+
+    try:
+        response = client.chat.completions.create(
+            model="llama-3.3-70b-versatile",
+            messages=[
+                {"role": "system", "content": system_prompt},
+                {"role": "user", "content": user_prompt}
+            ],
+            temperature=0.7,
+            max_tokens=4096
+        )
+        
+        content = response.choices[0].message.content.strip()
+        
+        # Clean up markdown code blocks
+        if content.startswith("```json"):
+            content = content[7:]
+        elif content.startswith("```"):
+            content = content[3:]
+        if content.endswith("```"):
+            content = content[:-3]
+        
+        content = content.strip()
+        content = clean_json_string(content)
+        
+        # Try to parse JSON
+        try:
+            data = json.loads(content)
+            return data
+        except json.JSONDecodeError as e:
+            # Try to fix common issues
+            content = re.sub(r'(".*?)\n(.*?")', r'\1\\n\2', content, flags=re.DOTALL)
+            content = re.sub(r'(".*?)\t(.*?")', r'\1\\t\2', content, flags=re.DOTALL)
+            data = json.loads(content)
+            return data
+            
+    except Exception as e:
+        return {"error": f"Failed to generate roadmap: {str(e)}"}
+
+def format_roadmark_for_display(roadmap_data):
+    """Format roadmap as HTML"""
+    if "error" in roadmap_data:
+        return f"<p style='color:red; padding:20px;'>{roadmap_data['error']}</p>", None
+    
+    try:
+        html_output = f"""
+        <div style="font-family: 'Segoe UI', sans-serif; max-width: 900px; margin: 0 auto;">
+            <div style="background: linear-gradient(135deg, #667eea 0%, #764ba2 100%); color: white; padding: 30px; border-radius: 15px; margin-bottom: 20px;">
+                <h1 style="margin: 0;">🎯 {roadmap_data.get('domain', 'Unknown')} Roadmap</h1>
+                <p>Level: {roadmap_data.get('level', 'N/A')} | Duration: {roadmap_data.get('duration', 'N/A')}</p>
+            </div>
+            <div style="background: #f8f9fa; padding: 20px; border-radius: 10px; margin-bottom: 20px;">
+                <h3>Overview</h3>
+                <p>{roadmap_data.get('overview', 'No overview provided').replace(chr(10), '<br>')}</p>
+            </div>
+        """
+        
+        for i, phase in enumerate(roadmap_data.get('phases', []), 1):
+            html_output += f"""
+            <div style="border: 2px solid #e9ecef; border-radius: 15px; margin-bottom: 20px; overflow: hidden;">
+                <div style="background: #667eea; color: white; padding: 15px;">
+                    <h3 style="margin:0;">Phase {i}: {phase.get('phase_name', 'Unnamed')}</h3>
+                    <p style="margin:5px 0 0 0;">Duration: {phase.get('duration_weeks', 'N/A')} weeks</p>
+                </div>
+                <div style="padding: 20px;">
+                    <p>{phase.get('description', '').replace(chr(10), '<br>')}</p>
+                    <h4>Topics:</h4>
+            """
+            
+            for topic in phase.get('topics', []):
+                html_output += f"<div style='background:#f8f9fa; padding:10px; margin:10px 0; border-radius:5px;'>"
+                html_output += f"<p style='margin:0;'><b>{topic.get('name', 'Unnamed')}</b></p>"
+                html_output += f"<ul style='margin:5px 0;'>"
+                for res in topic.get('resources', []):
+                    html_output += f"<li>{res}</li>"
+                html_output += f"</ul>"
+                html_output += f"<p style='background:#fff3cd; padding:5px; margin:5px 0; border-radius:3px;'><b>Project:</b> {topic.get('practice_project', 'N/A')}</p>"
+                html_output += f"</div>"
+            
+            if phase.get('milestones'):
+                html_output += f"<h4>Milestones:</h4><ul>"
+                for ms in phase['milestones']:
+                    html_output += f"<li>☐ {ms}</li>"
+                html_output += f"</ul>"
+            
+            html_output += "</div></div>"
+        
+        html_output += "</div>"
+        return html_output, json.dumps(roadmap_data)
+        
+    except Exception as e:
+        return f"<p style='color:red; padding:20px;'>Error formatting display: {str(e)}</p>", None
+
+def generate_pdf(roadmap_data, domain):
+    """Generate PDF file"""
+    try:
+        if isinstance(roadmap_data, str):
+            roadmap_data = json.loads(roadmap_data)
+        
+        if not roadmap_data or "error" in roadmap_data:
+            return None
+        
+        pdf = FPDF()
+        pdf.set_auto_page_break(auto=True, margin=15)
+        
+        def clean(text):
+            if text is None:
+                return ""
+            text = str(text)
+            return text.encode('latin-1', 'replace').decode('latin-1')
+        
+        # Title Page
+        pdf.add_page()
+        pdf.set_font("Arial", 'B', 24)
+        pdf.set_text_color(102, 126, 234)
+        pdf.cell(0, 20, clean(f"Learning Roadmap: {roadmap_data.get('domain', domain)}"), ln=True, align='C')
+        
+        pdf.set_font("Arial", '', 12)
+        pdf.set_text_color(100, 100, 100)
+        pdf.cell(0, 10, clean(f"Level: {roadmap_data.get('level', 'N/A')} | Duration: {roadmap_data.get('duration', 'N/A')}"), ln=True, align='C')
+        pdf.ln(10)
+        
+        # Overview
+        pdf.set_font("Arial", 'B', 16)
+        pdf.set_text_color(0, 0, 0)
+        pdf.cell(0, 10, "Overview", ln=True)
+        pdf.set_font("Arial", '', 11)
+        pdf.multi_cell(0, 6, clean(roadmap_data.get('overview', 'No overview')))
+        pdf.ln(5)
+        
+        # Phases
+        for i, phase in enumerate(roadmap_data.get('phases', []), 1):
+            pdf.add_page()
+            pdf.set_font("Arial", 'B', 18)
+            pdf.set_fill_color(102, 126, 234)
+            pdf.set_text_color(255, 255, 255)
+            pdf.cell(0, 12, clean(f"Phase {i}: {phase.get('phase_name', f'Phase {i}')}"), ln=True, fill=True)
+            
+            pdf.set_text_color(100, 100, 100)
+            pdf.set_font("Arial", 'I', 11)
+            pdf.cell(0, 8, f"Duration: {phase.get('duration_weeks', 'N/A')} weeks", ln=True)
+            pdf.ln(2)
+            
+            pdf.set_text_color(0, 0, 0)
+            pdf.set_font("Arial", '', 11)
+            pdf.multi_cell(0, 6, clean(phase.get('description', '')))
+            pdf.ln(5)
+            
+            pdf.set_font("Arial", 'B', 13)
+            pdf.set_text_color(102, 126, 234)
+            pdf.cell(0, 8, "Topics & Resources:", ln=True)
+            
+            for topic in phase.get('topics', []):
+                pdf.set_font("Arial", 'B', 11)
+                pdf.set_text_color(0, 0, 0)
+                pdf.cell(0, 6, clean(f"- {topic.get('name', 'Topic')}"), ln=True)
+                
+                pdf.set_font("Arial", '', 10)
+                pdf.set_text_color(80, 80, 80)
+                for resource in topic.get('resources', []):
+                    pdf.cell(10)
+                    pdf.cell(0, 5, clean(f"  * {resource}"), ln=True)
+                
+                pdf.set_text_color(200, 100, 0)
+                pdf.set_font("Arial", 'B', 10)
+                pdf.cell(0, 5, clean(f"Project: {topic.get('practice_project', 'N/A')}"), ln=True)
+                pdf.ln(3)
+            
+            if phase.get('milestones'):
+                pdf.set_font("Arial", 'B', 12)
+                pdf.set_text_color(40, 167, 69)
+                pdf.cell(0, 8, "Milestones:", ln=True)
+                pdf.set_font("Arial", '', 10)
+                pdf.set_text_color(0, 0, 0)
+                for milestone in phase.get('milestones', []):
+                    pdf.cell(10)
+                    pdf.cell(0, 5, clean(f"[ ] {milestone}"), ln=True)
+                pdf.ln(5)
+        
+        # Resources
+        if roadmap_data.get('essential_resources'):
+            pdf.add_page()
+            pdf.set_font("Arial", 'B', 18)
+            pdf.set_fill_color(102, 126, 234)
+            pdf.set_text_color(255, 255, 255)
+            pdf.cell(0, 12, "Essential Resources", ln=True, fill=True)
+            pdf.ln(5)
+            
+            resources = roadmap_data['essential_resources']
+            sections = [
+                ("Recommended Courses", resources.get('courses', [])), 
+                ("Books", resources.get('books', [])),
+                ("Communities", resources.get('communities', [])),
+                ("Tools", resources.get('tools', []))
+            ]
+            
+            for title, items in sections:
+                if items:
+                    pdf.set_font("Arial", 'B', 13)
+                    pdf.set_text_color(102, 126, 234)
+                    pdf.cell(0, 8, clean(title), ln=True)
+                    pdf.set_font("Arial", '', 10)
+                    pdf.set_text_color(0, 0, 0)
+                    for item in items:
+                        pdf.cell(5)
+                        pdf.cell(0, 5, clean(f"- {item}"), ln=True)
+                    pdf.ln(3)
+        
+        # Save to /tmp for Hugging Face
+        safe_domain = "".join(c for c in domain if c.isalnum() or c in (' ', '-', '_')).rstrip().replace(' ', '_') or "roadmap"
+        filename = f"roadmap_{safe_domain}_{datetime.now().strftime('%Y%m%d')}.pdf"
+        filepath = os.path.join("/tmp", filename)
+        pdf.output(filepath)
+        return filepath
+        
+    except Exception as e:
+        print(f"PDF Error: {e}")
+        return None
+
+# Build Gradio Interface
+def create_app():
+    with gr.Blocks(title="AI Learning Roadmap Generator") as app:
+        roadmap_json = gr.State("")
+        domain_name = gr.State("")
+        
+        gr.Markdown("# 🎯 AI Learning Roadmap Generator")
+        gr.Markdown("Generate personalized learning paths for any skill domain")
+        
+        with gr.Row():
+            with gr.Column(scale=1):
+                domain_input = gr.Textbox(label="Learning Domain", placeholder="e.g., Machine Learning, Web Development...")
+                level_input = gr.Dropdown(["Beginner", "Intermediate", "Advanced"], label="Level", value="Beginner")
+                
+                with gr.Row():
+                    time_input = gr.Number(value=3, minimum=1, label="Time")
+                    time_unit = gr.Dropdown(["Weeks", "Months", "Years"], value="Months", label="Unit")
+                
+                generate_btn = gr.Button("🚀 Generate Roadmap", variant="primary")
+                
+                gr.Examples(
+                    [["Machine Learning", "Beginner", 6, "Months"],
+                     ["Web Development", "Intermediate", 4, "Months"],
+                     ["Python Programming", "Beginner", 8, "Weeks"]],
+                    inputs=[domain_input, level_input, time_input, time_unit]
+                )
+            
+            with gr.Column(scale=2):
+                html_output = gr.HTML(label="Your Roadmap")
+                
+                with gr.Row():
+                    download_btn = gr.Button("📄 Download PDF", variant="primary", visible=False)
+                
+                pdf_output = gr.File(label="Your PDF", interactive=False, visible=False)
+        
+        def generate(domain, level, time_val, unit):
+            if not domain or not domain.strip():
+                return "Please enter a domain", gr.update(visible=False), gr.update(visible=False), "", ""
+            
+            result = generate_roadmap(domain, level, time_val, unit)
+            
+            if "error" in result:
+                return f"<p style='color:red; padding:20px;'>❌ {result['error']}</p>", gr.update(visible=False), gr.update(visible=False), "", ""
+            
+            html, json_str = format_roadmark_for_display(result)
+            if json_str is None:
+                return html, gr.update(visible=False), gr.update(visible=False), "", ""
+            
+            return html, gr.update(visible=True), gr.update(visible=True), json_str, domain
+        
+        def download(json_data, dom):
+            if not json_data:
+                return None
+            return generate_pdf(json_data, dom)
+        
+        generate_btn.click(
+            fn=generate,
+            inputs=[domain_input, level_input, time_input, time_unit],
+            outputs=[html_output, download_btn, pdf_output, roadmap_json, domain_name]
+        )
+        
+        download_btn.click(
+            fn=download,
+            inputs=[roadmap_json, domain_name],
+            outputs=pdf_output
+        )
+    
+    return app
+
+app = create_app()
+app.launch()