Spaces:
Running
Running
File size: 5,435 Bytes
8c52bbf cfe4dff 8c52bbf e1dd757 be4e070 8c52bbf 0d9e5b3 be4e070 0d9e5b3 8c52bbf 0d9e5b3 8c52bbf be4e070 4a33878 be4e070 0d9e5b3 8c52bbf be4e070 8c52bbf 0d9e5b3 e1dd757 0d9e5b3 be4e070 0d9e5b3 be4e070 0d9e5b3 e1dd757 0d9e5b3 be4e070 e1dd757 be4e070 e1dd757 be4e070 0d9e5b3 be4e070 0d9e5b3 be4e070 0d9e5b3 be4e070 0d9e5b3 be4e070 0d9e5b3 be4e070 8f4ea34 be4e070 8f4ea34 be4e070 8f4ea34 be4e070 8f4ea34 be4e070 8f4ea34 0d9e5b3 8f4ea34 be4e070 8f4ea34 be4e070 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 102 103 104 105 106 107 108 109 110 111 112 113 114 115 116 117 118 119 120 121 122 123 124 125 126 127 128 129 130 131 132 133 134 135 136 137 138 139 140 141 142 143 144 145 146 147 |
import streamlit as st
import ast
import base64
import streamlit.components.v1 as components
from transformers import pipeline
from gtts import gTTS
import os
st.set_page_config(page_title="AR/VR Code Visualizer", layout="wide")
st.title("👓 AR/VR Code Visualizer with Editing, Interaction, and Export")
@st.cache_resource
def load_model():
return pipeline("summarization", model="philschmid/bart-large-cnn-samsum")
summarizer = load_model()
# In-browser code editor
st.subheader("📝 Write or Paste Your Python Code")
code = st.text_area("Enter your Python code here", height=300)
if code.strip():
st.code(code, language="python")
# Parse AST for functions and calls
tree = ast.parse(code)
class FunctionCallVisitor(ast.NodeVisitor):
def __init__(self):
self.calls = {}
def visit_FunctionDef(self, node):
caller = node.name
self.calls[caller] = []
for child in ast.walk(node):
if isinstance(child, ast.Call) and isinstance(child.func, ast.Name):
self.calls[caller].append(child.func.id)
self.generic_visit(node)
visitor = FunctionCallVisitor()
visitor.visit(tree)
call_graph = visitor.calls
all_functions = list(call_graph.keys())
st.subheader("📊 Function Calls")
for fn, callees in call_graph.items():
st.write(f"🔹 `{fn}` calls: {', '.join(callees) if callees else 'None'}")
# Generate AI summary
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)
summary_text = summary[0]['summary_text']
st.success(summary_text)
# Generate voice narration
st.subheader("🔊 Voice Narration")
tts = gTTS(text=summary_text)
tts.save("summary.mp3")
st.audio("summary.mp3", format="audio/mp3")
# A-Frame VR scene with interactivity and export
def generate_aframe(call_graph):
spacing = 3
boxes = []
lines = []
positions = []
function_positions = {}
i = 0
for fn in call_graph:
x = i * spacing
function_positions[fn] = (x, 1, -3)
positions.append(fn)
boxes.append(f"""
<a-box id="{fn}" position="{x} 1 -3" depth="0.5" height="0.5" width="2" color="#FFC65D"
class="clickable"
event-set__enter="_event: mouseenter; color: #00CED1"
event-set__leave="_event: mouseleave; color: #FFC65D"
onclick="say('{fn}')">
</a-box>
<a-text value="{fn}" position="{x} 2 -3" align="center" color="#000"></a-text>
""")
i += 1
for caller, callees in call_graph.items():
for callee in callees:
if callee in function_positions:
x1, y1, z1 = function_positions[caller]
x2, y2, z2 = function_positions[callee]
lines.append(f"""
<a-entity line="start: {x1} {y1} {z1}; end: {x2} {y2} {z2}; color: red"></a-entity>
""")
# JavaScript for text-to-speech on click + screenshot
js = """
<script>
function say(text) {
const utter = new SpeechSynthesisUtterance(text);
speechSynthesis.speak(utter);
}
function downloadScreenshot() {
const scene = document.querySelector('a-scene');
scene.components.screenshot.capture('perspective');
setTimeout(() => {
const a = document.createElement('a');
a.href = scene.components.screenshot.canvas.toDataURL();
a.download = 'scene.png';
a.click();
}, 500);
}
</script>
"""
html = f"""
<!DOCTYPE html>
<html>
<head>
<script src="https://aframe.io/releases/1.3.0/aframe.min.js"></script>
<script src="https://cdn.jsdelivr.net/gh/donmccurdy/[email protected]/dist/aframe-extras.min.js"></script>
<script src="https://unpkg.com/[email protected]/dist/aframe-screenshot-component.min.js"></script>
{js}
</head>
<body>
<button onclick="downloadScreenshot()" style="position: absolute; z-index: 9999; top: 10px; left: 10px; font-size: 16px;">📸 Download Screenshot</button>
<a-scene screenshot>
<a-entity position="0 1.6 3">
<a-camera wasd-controls-enabled="true" look-controls-enabled="true"></a-camera>
</a-entity>
<a-light type="ambient" color="#FFF"></a-light>
<a-plane rotation="-90 0 0" width="40" height="40" color="#7BC8A4"></a-plane>
{''.join(boxes)}
{''.join(lines)}
</a-scene>
</body>
</html>
"""
return html
aframe_html = generate_aframe(call_graph)
b64 = base64.b64encode(aframe_html.encode()).decode()
data_url = f"data:text/html;base64,{b64}"
st.subheader("🌐 Interactive 3D Function Visualizer")
components.iframe(data_url, height=600)
else:
st.info("Write some Python code above to visualize, narrate, and explore it in VR.")
|