Update app.py

app.py

@@ -4,33 +4,21 @@ import streamlit.components.v1 as components
 # Set page config for dark theme
 st.set_page_config(layout="wide", page_title="Web UI", page_icon=":ghost:", initial_sidebar_state="expanded")
 
-# Apply dark theme
+# Apply dark theme and maximize iframe
 st.markdown("""
 <style>
-    .reportview-container {
-        background-color: #000000;
-    }
-    .sidebar .sidebar-content {
-        background-color: #262730;
-    }
-    .Widget>label {
-        color: white;
-    }
-    .stRadio>div{
-        color: white;
-    }
-    .stCheckbox>div{
-        color: white;
-    }
-    .stSelectbox>div{
-        color: white;
-    }
-    .main .block-container {
-        background-color: #000000;
-    }
-    iframe {
-        background-color: #000000;
-    }
+    #root > div:nth-child(1) > div > div > div > div > section > div {padding-top: 0rem;}
+    .reportview-container {background-color: #000000;}
+    .sidebar .sidebar-content {background-color: #262730;}
+    .Widget>label {color: white;}
+    .stRadio>div{color: white;}
+    .stCheckbox>div{color: white;}
+    .stSelectbox>div{color: white;}
+    .main .block-container {background-color: #000000; padding: 0rem;}
+    iframe {width: 100vw; height: 100vh; border: none;}
+    header {visibility: hidden;}
+    #MainMenu {visibility: hidden;}
+    footer {visibility: hidden;}
 </style>
 """, unsafe_allow_html=True)
 
@@ -40,35 +28,28 @@ menu = st.sidebar.radio("선택하세요:", ("A", "B"))
 
 # Main content
 if menu == "A":
-    components.iframe("https://seawolf2357-flxloraexp.hf.space", height=800, scrolling=True)
+    components.iframe("https://seawolf2357-flxloraexp.hf.space", height=1000, scrolling=True)
 elif menu == "B":
-    components.iframe("http://hugpu.ai:8000", height=800, scrolling=True)
+    try:
+        components.iframe("http://hugpu.ai:8000", height=1000, scrolling=True)
+    except Exception as e:
+        st.error(f"B 메뉴 로딩 중 오류 발생: {str(e)}")
+        st.info("http://hugpu.ai:8000 에 접근할 수 없습니다. URL이 올바른지, 그리고 공개적으로 접근 가능한지 확인해 주세요.")
 
-# Make the layout responsive
-st.markdown("""
-<style>
-    .reportview-container .main .block-container {
-        max-width: 1000px;
-        padding-top: 2rem;
-        padding-right: 2rem;
-        padding-left: 2rem;
-        padding-bottom: 2rem;
-    }
-    .sidebar .sidebar-content {
-        width: 300px;
-    }
-</style>
-""", unsafe_allow_html=True)
-
-# Ensure iframes are loaded correctly
+# Make the layout responsive and ensure iframes are loaded correctly
 st.markdown("""
 <script>
     window.addEventListener('load', function() {
         var iframes = document.getElementsByTagName('iframe');
         for (var i = 0; i < iframes.length; i++) {
-            iframes[i].onload = function() {
-                this.style.height = this.contentWindow.document.body.scrollHeight + 'px';
-            }
+            iframes[i].style.height = window.innerHeight + 'px';
+            iframes[i].style.width = '100%';
+        }
+    });
+    window.addEventListener('resize', function() {
+        var iframes = document.getElementsByTagName('iframe');
+        for (var i = 0; i < iframes.length; i++) {
+            iframes[i].style.height = window.innerHeight + 'px';
         }
     });
 </script>