Spaces:
Build error
Build error
| import re | |
| from typing import Any | |
| import streamlit as st | |
| import types | |
| from streamlit_card import card | |
| class Render(): | |
| SOURCE_CARD={ | |
| "card": {"width": "100%","height":"60px","margin":"0px","borderRadius": "3px","box-shadow":"0px 0px 0px rgba(0, 0, 0, 0)"}, | |
| "title":{"font-size":"0.9em","text-align":"left","width":"90%"}, | |
| "text":{"text-align":"left","font-size":"0.6em","text-align":"left","width":"90%"}, | |
| "filter": {"background-color": "rgba(50, 50, 50, 0.5)"} | |
| } | |
| SUGGESTION_CARD={ | |
| "card": {"width": "100%","height":"60px","margin":"0px","borderRadius": "3px","box-shadow":"0px 0px 0px rgba(0, 0, 0, 0)"}, | |
| "title":{"font-size":"0.9em","text-align":"left","width":"90%"}, | |
| "filter": {"background-color": "rgba(50, 50, 50, 0.5)"} | |
| } | |
| CARD_NUM = 3 | |
| def __init__(self) -> None: | |
| pass | |
| def __call__(self, answer=None): | |
| print(st.session_state) | |
| self.render_suggestion() | |
| self.render_source() | |
| self.render_answer(answer=answer) | |
| def render_suggestion(self): | |
| suggestion = st.session_state.suggestion | |
| if not suggestion: | |
| return | |
| cols = st.columns([2,2,2,2,2,2],gap="small") | |
| for idx,item in enumerate(suggestion): | |
| name = item.get('name','') | |
| url = item.get('url','') | |
| if name == "" or url == "": | |
| continue | |
| with cols[idx]: | |
| try: | |
| print(f"suggestion_{idx}") | |
| card( | |
| key=f"suggestion_{idx}", | |
| title=name, | |
| text='', | |
| styles=Render.SUGGESTION_CARD, | |
| url=url, | |
| ) | |
| except Exception as e: | |
| pass | |
| def render_answer(self,answer): | |
| if st.session_state.answer: | |
| st.subheader("Answer") | |
| text_container = st.empty() | |
| text_container.empty() | |
| text_container.write(st.session_state.answer) | |
| return | |
| if not answer: | |
| return | |
| st.subheader("Answer") | |
| text_container = st.empty() | |
| text = "" | |
| for data in answer: | |
| text += data | |
| text_container.empty() | |
| text_container.write(text) | |
| st.session_state.answer = text | |
| def render_source(self): | |
| source = st.session_state.source | |
| if not source: | |
| return | |
| st.subheader("Source") | |
| cols = st.columns([2,2,2,2],gap="small") | |
| for idx,item in enumerate(source): | |
| title = item.get("name","") | |
| title = re.search(r"^(.{0,60}\S\b|)", title).group(0) | |
| text = item.get("description","") | |
| text = re.search(r"^(.{0,120}\S\b|)", text).group(0)+" ..." | |
| url = item.get("url","") | |
| if title == "" or text == "" or url == "": | |
| continue | |
| with cols[idx]: | |
| print(f"source_{idx}") | |
| card( | |
| key=f"source_{idx}", | |
| title=title, | |
| text='', | |
| styles=Render.SOURCE_CARD, | |
| url=url, | |
| ) |