Update app.py

app.py

@@ -3,14 +3,7 @@ import ast
 import streamlit.components.v1 as components
 from transformers import pipeline
 
-# --- App title ---
-st.set_page_config(page_title="AR/VR Enhanced Code Learning", layout="wide")
-st.title("👓 AR/VR Enhanced Code Visualizer")
-
-# --- File upload ---
-uploaded_file = st.file_uploader("Upload your Python file", type=["py"])
-
-# --- Hugging Face model ---
+# --- Hugging Face summarization model ---
 @st.cache_resource
 def load_model():
     summarizer = pipeline("summarization", model="philschmid/bart-large-cnn-samsum")
@@ -18,33 +11,42 @@ def load_model():
 
 summarizer = load_model()
 
-# --- Process file ---
+# --- Streamlit page config ---
+st.set_page_config(page_title="AR/VR Code Visualizer", layout="wide")
+st.title("👓 AR/VR Enhanced Code Visualizer")
+
+# --- File uploader ---
+uploaded_file = st.file_uploader("📁 Upload your Python file", type=["py"])
+
 if uploaded_file is not None:
     code = uploaded_file.read().decode('utf-8')
     st.subheader("📄 Uploaded Code")
     st.code(code, language='python')
 
-    # --- Parse AST ---
+    # --- AST Parsing ---
     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 Code Elements")
-    st.write(f"**Functions:** {functions}")
-    st.write(f"**Classes:** {classes}")
+    st.subheader("🔍 Detected Elements")
+    st.write(f"**Functions:** {functions if functions else 'None found'}")
+    st.write(f"**Classes:** {classes if classes else 'None found'}")
 
     # --- Hugging Face Summary ---
-    st.subheader("🧠 AI Explanation of Code")
-    input_text = f"This Python code contains functions: {functions} and classes: {classes}. Explain their purpose simply."
-    summary = summarizer(input_text, max_length=60, min_length=10, do_sample=False)
-    st.write(summary[0]['summary_text'])
-
-    # --- AR/VR Viewer ---
-    st.subheader("🌐 AR/VR Visualization (Demo)")
-
-    st.info("This is a sample AR/VR scene visualizing functions/classes. Future versions will show dynamic graphs of your code!")
-
-    aframe_url = "https://aframe.io/aframe/examples/showcase/helloworld/"  # Placeholder demo scene
-
+    st.subheader("🧠 Code Explanation")
+    prompt = f"This Python code has functions: {functions} and classes: {classes}. Explain what the code might be doing."
+    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!")
+
+    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.")