ys/GenericAgent
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

fix: compatible with Streamlit 1.56+ iframe API change

Browse Source
This commit is contained in:
Liang Jiaqing
2026-04-16 17:48:58 +08:00
parent 9e18ce26dc
commit 47bcb1b01a
2 changed files with 13 additions and 5 deletions
Download Patch File Download Diff File Expand all files Collapse all files

8
frontends/stapp.py
View File

@@ -146,7 +146,11 @@ for msg in st.session_state.messages:
# Scroll-height ghost fix: during streaming, expander open/close mid-animation can leave # Scroll-height ghost fix: during streaming, expander open/close mid-animation can leave
# phantom height → scrollbar long but can't scroll to bottom. Periodically detect & reflow. # phantom height → scrollbar long but can't scroll to bottom. Periodically detect & reflow.
import streamlit.components.v1 as components try:
from streamlit import iframe as _st_iframe # 1.56+
_embed_html = lambda html, **kw: _st_iframe(html, **{k: max(v, 1) if isinstance(v, int) else v for k, v in kw.items()})
except (ImportError, AttributeError):
from streamlit.components.v1 import html as _embed_html # ≤1.55
_js_scroll_fix = ("!function(){var p=window.parent;if(p.__sfx)return;p.__sfx=1;" _js_scroll_fix = ("!function(){var p=window.parent;if(p.__sfx)return;p.__sfx=1;"
"var d=p.document;setInterval(function(){" "var d=p.document;setInterval(function(){"
"var m=d.querySelector('section.main');if(!m)return;" "var m=d.querySelector('section.main');if(!m)return;"
@@ -165,7 +169,7 @@ _js_ime_fix = ("" if os.name == 'nt' else
"e.key==='Enter'&&!e.shiftKey&&(e.isComposing||c||e.keyCode===229)&&" "e.key==='Enter'&&!e.shiftKey&&(e.isComposing||c||e.keyCode===229)&&"
"(e.stopImmediatePropagation(),e.preventDefault())},!0))})}" "(e.stopImmediatePropagation(),e.preventDefault())},!0))})}"
"f();new MutationObserver(f).observe(d.body,{childList:1,subtree:1})}()") "f();new MutationObserver(f).observe(d.body,{childList:1,subtree:1})}()")
components.html(f'<script>{_js_scroll_fix};{_js_ime_fix}</script>', height=0) _embed_html(f'<script>{_js_scroll_fix};{_js_ime_fix}</script>', height=0)
if prompt := st.chat_input("请输入指令"): if prompt := st.chat_input("请输入指令"):
st.session_state.messages.append({"role": "user", "content": prompt}) st.session_state.messages.append({"role": "user", "content": prompt})

10
frontends/stapp2.py
View File

@@ -9,7 +9,11 @@ except: pass
sys.path.append(os.path.abspath(os.path.join(os.path.dirname(__file__), '..'))) sys.path.append(os.path.abspath(os.path.join(os.path.dirname(__file__), '..')))
import streamlit as st import streamlit as st
import streamlit.components.v1 as components try:
from streamlit import iframe as _st_iframe # 1.56+
_embed_html = lambda html, **kw: _st_iframe(html, **{k: max(v, 1) if isinstance(v, int) else v for k, v in kw.items()})
except (ImportError, AttributeError):
from streamlit.components.v1 import html as _embed_html # ≤1.55
import time, json, re, threading, queue import time, json, re, threading, queue
from datetime import datetime from datetime import datetime
from agentmain import GeneraticAgent from agentmain import GeneraticAgent
@@ -951,8 +955,8 @@ init_session_state()
# Inject Anthropic theme # Inject Anthropic theme
st.markdown(ANTHROPIC_CSS, unsafe_allow_html=True) st.markdown(ANTHROPIC_CSS, unsafe_allow_html=True)
st.markdown(build_dynamic_font_css(110.0), unsafe_allow_html=True) st.markdown(build_dynamic_font_css(110.0), unsafe_allow_html=True)
components.html(ANTHROPIC_SELECTBOX_SCRIPT, height=0, width=0) _embed_html(ANTHROPIC_SELECTBOX_SCRIPT, height=0, width=0)
components.html(build_header_agent_badge_script(), height=0, width=0) _embed_html(build_header_agent_badge_script(), height=0, width=0)
st.session_state.agent_name = 'Generic Agent' st.session_state.agent_name = 'Generic Agent'
with st.chat_message("assistant"): with st.chat_message("assistant"):