fix(streamlit): migrate deprecated html embeds and harden AI roast flow

- Replace deprecated st.components.v1.html usage with iframe-based renderer

- Remove obsolete components import

- Reuse loaded profile data in AI roast to avoid secondary 403 failures

- Normalize roast profile payload shape for language data

- Add IssueCommentEvent and PullRequestReviewCommentEvent to known event types

Author: devanshi14malhotra

app.py

@@ -6,7 +6,6 @@
 
 # HEX color regex validation pattern
 HEX_COLOR_REGEX = re.compile(r'^#[0-9A-Fa-f]{6}$')
-import streamlit.components.v1 as components
 from dotenv import load_dotenv
 from config.settings import get_settings
 from roast_widget_streamlit import render_roast_widget
@@ -413,6 +412,12 @@ def show_code_area(code_content, label="Markdown Code"):
     st.markdown(f"**{label}** (Copy below)")
     st.text_area(label, value=code_content, height=100, label_visibility="collapsed")
 
+
+def render_embedded_html(html_content: str, *, height: int) -> None:
+    """Render inline HTML via iframe to avoid deprecated st.components.v1.html."""
+    html_b64 = base64.b64encode(html_content.encode("utf-8")).decode("ascii")
+    st.iframe(f"data:text/html;base64,{html_b64}", height=height, scrolling=False)
+
 def render_tab(svg_bytes, endpoint, username, selected_theme, custom_colors, hide_params=None, code_template=None, excluded_languages=None, output_format="Markdown", font_override=None, extra_params=None):
     col1, col2 = st.columns([1.5, 1])
     with col1:
@@ -433,7 +438,7 @@ def render_tab(svg_bytes, endpoint, username, selected_theme, custom_colors, hid
         svg_b64 = base64.b64encode(svg_bytes.encode("utf-8")).decode("utf-8")
         filename_prefix_safe = json.dumps(f"{endpoint}_{username}")
 
-        components.html(f"""
+        render_embedded_html(f"""
         <div style="display:flex; flex-direction:column; gap:8px; margin-top:4px;">
             <button onclick="downloadSVGAs('png')" style="
                 width:100%; padding:8px; font-size:14px; cursor:pointer;
@@ -619,7 +624,7 @@ def render_tab(svg_bytes, endpoint, username, selected_theme, custom_colors, hid
 
     with spark_col1:
         # We wrap it in a component so the animations and styles actually trigger
-        st.components.v1.html(f"""
+        render_embedded_html(f"""
             <div style="background: {current_theme_opts.get('bg_color', '#0d1117')}; border-radius: 12px; border: 1px solid {current_theme_opts.get('border_color', '#30363d')}; overflow: hidden; position: relative; margin-bottom: 5px;">
                 <!-- Corner Accents -->
                 <div style="position: absolute; top: 0; left: 0; width: 10px; height: 10px; border-top: 2px solid {theme_color}; border-left: 2px solid {theme_color};"></div>
@@ -884,7 +889,7 @@ def render_tab(svg_bytes, endpoint, username, selected_theme, custom_colors, hid
     st.markdown("Let AI roast your GitHub profile with humor!")
 
     if username:
-        render_roast_widget(username)
+        render_roast_widget(username, profile_data=data)
     else:
         st.warning("Please enter a GitHub username in the sidebar.")
 

roast_widget_streamlit.py

@@ -4,12 +4,41 @@
 """
 
 import streamlit as st
-import os
+from typing import Any
 from ai.ai_roast_service import generate_profile_roast
 from utils.github_utils import fetch_github_stats
 
 
-def render_roast_widget(username: str):
+def _normalize_profile_data(profile_data: dict[str, Any] | None, username: str) -> dict[str, Any]:
+    """Normalize profile shape so AI roast service always gets expected keys."""
+    base: dict[str, Any] = {
+        "username": username,
+        "public_repos": 0,
+        "total_commits": 0,
+        "top_languages": [],
+    }
+    if not profile_data:
+        return base
+
+    normalized = dict(base)
+    normalized["username"] = profile_data.get("username") or profile_data.get("login") or username
+    normalized["public_repos"] = profile_data.get("public_repos", 0)
+    normalized["total_commits"] = profile_data.get("total_commits", 0)
+
+    raw_langs = profile_data.get("top_languages", [])
+    langs: list[dict[str, Any]] = []
+    for entry in raw_langs:
+        if isinstance(entry, dict):
+            name = entry.get("name")
+            if name:
+                langs.append({"name": name, "count": entry.get("count", 0)})
+        elif isinstance(entry, (list, tuple)) and entry:
+            langs.append({"name": str(entry[0]), "count": entry[1] if len(entry) > 1 else 0})
+    normalized["top_languages"] = langs
+    return normalized
+
+
+def render_roast_widget(username: str, profile_data: dict[str, Any] | None = None):
     """
     Render the AI Roast widget in Streamlit
     
@@ -90,15 +119,19 @@ def render_roast_widget(username: str):
             if st.button("🎭 Generate Roast", use_container_width=True, type="primary"):
                 with st.spinner("🔥 Cooking up a roast..."):
                     try:
-                        # Fetch GitHub stats
-                        profile_data = fetch_github_stats(username)
-                        
-                        if profile_data:
+                        # Reuse already loaded profile data first to avoid rate-limit failures.
+                        source_profile = _normalize_profile_data(profile_data, username)
+                        if source_profile.get("total_commits", 0) <= 0 and not source_profile.get("top_languages"):
+                            fetched_profile = fetch_github_stats(username)
+                            if fetched_profile:
+                                source_profile = _normalize_profile_data(fetched_profile, username)
+
+                        if source_profile:
                             # Generate roast
-                            roast_result = generate_profile_roast(profile_data)
+                            roast_result = generate_profile_roast(source_profile)
                             st.session_state.roast_data = {
                                 'roast': roast_result['roast'],
-                                'profile': profile_data,
+                                'profile': source_profile,
                                 'source': roast_result['source']
                             }
                         else:

utils/api_validators.py

@@ -89,7 +89,8 @@ class GitHubEvent(BaseModel):
     def validate_type(cls, v):
         allowed_types = {
             'PullRequestEvent', 'IssuesEvent', 'PushEvent', 'CreateEvent',
-            'DeleteEvent', 'ForkEvent', 'WatchEvent', 'ReleaseEvent'
+            'DeleteEvent', 'ForkEvent', 'WatchEvent', 'ReleaseEvent',
+            'IssueCommentEvent', 'PullRequestReviewCommentEvent'
         }
         if v not in allowed_types:
             # Log unknown type but don't fail validation