Update app.py

app.py

@@ -3,9 +3,10 @@ import ast
 import base64
 import streamlit.components.v1 as components
 from transformers import pipeline
+from gtts import gTTS
 
 st.set_page_config(page_title="AR/VR Code Visualizer", layout="wide")
-st.title("👓 AR/VR Code Visualizer with Call Relationships")
+st.title("👓 AR/VR Code Visualizer with Call Relationships + Narration")
 
 @st.cache_resource
 def load_model():
@@ -19,13 +20,12 @@ if uploaded_file:
     code = uploaded_file.read().decode("utf-8")
     st.code(code, language="python")
 
-    # Parse AST
+    # Parse AST for functions and call relationships
     tree = ast.parse(code)
 
-    # Find functions and their calls
     class FunctionCallVisitor(ast.NodeVisitor):
         def __init__(self):
-            self.calls = {}  # {caller: [callee1, callee2]}
+            self.calls = {}
 
         def visit_FunctionDef(self, node):
             caller = node.name
@@ -40,24 +40,31 @@ if uploaded_file:
     call_graph = visitor.calls
     all_functions = list(call_graph.keys())
 
-    # Show extracted relationships
+    # Show call relationships
     st.subheader("📊 Call Relationships")
     for fn, callees in call_graph.items():
         st.write(f"🔹 `{fn}` calls: {', '.join(callees) if callees else 'None'}")
 
-    # Summarize using Hugging Face
+    # Hugging Face summarization
     prompt = f"Explain the structure and purpose of the following functions and how they call each other: {call_graph}"
     summary = summarizer(prompt, max_length=60, min_length=15, do_sample=False)
-    st.success(summary[0]['summary_text'])
+    summary_text = summary[0]['summary_text']
+    st.success(summary_text)
 
-    # Generate A-Frame HTML
+    # Voice narration
+    st.subheader("🔊 Voice Narration of Code Summary")
+    tts = gTTS(text=summary_text)
+    tts.save("summary.mp3")
+    audio_file = open("summary.mp3", "rb")
+    st.audio(audio_file.read(), format="audio/mp3")
+
+    # Generate A-Frame scene
     def generate_aframe(call_graph):
         spacing = 3
         boxes = []
         lines = []
         positions = {}
 
-        # Assign fixed positions on x-axis
         for i, fn in enumerate(call_graph):
             x = i * spacing
             positions[fn] = (x, 1, -3)
@@ -66,7 +73,6 @@ if uploaded_file:
                 <a-text value="{fn}" position="{x} 1.8 -3" align="center" color="#000"></a-text>
             ''')
 
-        # Draw arrows between caller → callee
         for caller, callees in call_graph.items():
             for callee in callees:
                 if callee in positions:
@@ -105,4 +111,4 @@ if uploaded_file:
     components.iframe(data_url, height=500)
 
 else:
-    st.info("Upload a Python file to visualize function call relationships.")
+    st.info("Upload a Python file to visualize its structure, relationships, and get narrated summary.")