Spaces:
Running
Running
File size: 3,147 Bytes
3480bcd |
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 |
import streamlit as st
from typing import Dict, Any
from MiniManus import manager_agent
from pathlib import Path
st.set_page_config(
page_title="Manus AI Assistant",
page_icon="🤖",
layout="centered",
initial_sidebar_state="collapsed"
)
# Initialize session state
if "messages" not in st.session_state:
st.session_state.messages = []
if "total_queries" not in st.session_state:
st.session_state.total_queries = 0
def call_manager_agent(query: str):
"""Call the manager agent and return the result"""
try:
result = manager_agent.run(query)
return result
except Exception as e:
return f"❌ Error: {str(e)}"
def render_output(content: Any):
"""Render agent output automatically depending on type"""
# Case 1: Text / Markdown
if isinstance(content, str):
if content.strip().startswith("http"):
st.markdown(f"[🔗 Link]({content})")
elif Path(content).suffix in [".zip", ".pkl", ".csv", ".xlsx", ".json", ".txt", ".py"]:
# File path returned
file_path = Path(content)
if file_path.exists():
with open(file_path, "rb") as f:
st.download_button(
label=f"📂 Download {file_path.name}",
data=f,
file_name=file_path.name
)
else:
st.warning(f"⚠️ File not found: {file_path}")
else:
st.markdown(content)
# Case 2: Images
elif "PIL" in str(type(content)):
st.image(content, caption="Generated Image", use_container_width=True)
# Case 3: Matplotlib figures
elif "matplotlib" in str(type(content)):
st.pyplot(content)
# Case 4: Dict / JSON-like (show nicely)
elif isinstance(content, dict):
st.json(content)
# Case 5: List of results (recursively render)
elif isinstance(content, list):
for item in content:
render_output(item)
# Fallback
else:
st.write(content)
def display_message(message: Dict[str, Any]):
"""Display chat messages with auto-rendered assistant outputs"""
if message["role"] == "user":
st.chat_message("user").write(message["content"])
else:
with st.chat_message("assistant"):
render_output(message["content"])
def main():
"""Main chatbot loop"""
st.title("🤖 Manus AI Assistant")
for message in st.session_state.messages:
display_message(message)
user_input = st.chat_input("Ask Manus AI...")
if user_input:
st.session_state.messages.append({"role": "user", "content": user_input})
st.session_state.total_queries += 1
with st.chat_message("assistant"):
with st.spinner("Thinking..."):
response = call_manager_agent(user_input)
st.session_state.messages.append({"role": "assistant", "content": response})
render_output(response)
if __name__ == "__main__":
main() |