Update app.py

app.py

@@ -1,9 +1,10 @@
 import streamlit as st
 import ast
+import base64
 import streamlit.components.v1 as components
 from transformers import pipeline
 
-# --- Hugging Face summarization model ---
+# --- Load summarizer model ---
 @st.cache_resource
 def load_model():
     summarizer = pipeline("summarization", model="philschmid/bart-large-cnn-samsum")
@@ -11,42 +12,41 @@ def load_model():
 
 summarizer = load_model()
 
-# --- Streamlit page config ---
 st.set_page_config(page_title="AR/VR Code Visualizer", layout="wide")
 st.title("👓 AR/VR Enhanced Code Visualizer")
 
-# --- File uploader ---
+# --- Upload Python file ---
 uploaded_file = st.file_uploader("📁 Upload your Python file", type=["py"])
 
+# --- Parse code and show output ---
 if uploaded_file is not None:
     code = uploaded_file.read().decode('utf-8')
     st.subheader("📄 Uploaded Code")
     st.code(code, language='python')
 
-    # --- AST Parsing ---
+    # Parse AST
     tree = ast.parse(code)
     functions = [node.name for node in ast.walk(tree) if isinstance(node, ast.FunctionDef)]
     classes = [node.name for node in ast.walk(tree) if isinstance(node, ast.ClassDef)]
 
-    st.subheader("🔍 Detected Elements")
-    st.write(f"**Functions:** {functions if functions else 'None found'}")
-    st.write(f"**Classes:** {classes if classes else 'None found'}")
+    st.subheader("🔍 Code Structure")
+    st.write(f"**Functions:** {functions or 'None'}")
+    st.write(f"**Classes:** {classes or 'None'}")
 
-    # --- Hugging Face Summary ---
-    st.subheader("🧠 Code Explanation")
-    prompt = f"This Python code has functions: {functions} and classes: {classes}. Explain what the code might be doing."
+    # Hugging Face Summarization
+    st.subheader("🧠 Code Explanation (AI)")
+    prompt = f"This code contains functions: {functions} and classes: {classes}. Explain what this code does."
     try:
         summary = summarizer(prompt, max_length=60, min_length=15, do_sample=False)
         st.success(summary[0]['summary_text'])
     except Exception as e:
         st.error(f"Summarization failed: {e}")
 
-    # --- AR/VR Scene ---
-    st.subheader("🌐 AR/VR Visualization")
-    st.info("Below is a sample 3D scene visualizing code elements. Dynamic visual generation coming soon!")
+    # Build dynamic A-Frame HTML
+    def generate_aframe_html(functions, classes):
+        elements = []
+        x_pos = -3
 
-    aframe_url = "https://your-username.github.io/ar-vr-code-visualizer/code_visualizer.html"  # replace with your actual GitHub Pages link
-    components.iframe(aframe_url, height=500)
-
-else:
-    st.info("Upload a Python file to begin visualizing and explaining it.")
+        for fn in functions:
+            elements.append(f"""
+                <a-box