Update src/streamlit_app.py

src/streamlit_app.py

@@ -7,118 +7,106 @@ import streamlit as st
 import numpy as np
 import pandas as pd
 from gtts import gTTS
-# assitant 아바타 이미지 경로
-assistant_icon_path = "https://huggingface.co/spaces/randmimc/aitom/resolve/main/chat_icon/assistant.ico"
 
-# 세션 상태에서 approval_state 및 messages 초기화
-if "approval_state" not in st.session_state:
-    st.session_state.approval_state = "require"
-if "messages" not in st.session_state:
-    st.session_state.messages = []
+# 예시 데이터
+options = ["", "2024-08-24", "2024-09-26"]
+df = pd.read_csv("https://huggingface.co/spaces/randmimc/mobility_condition/raw/main/chat_icon/text.csv", on_bad_lines='skip') 
+text_input = "위의 증상들이 보일 경우, 다음과 같은 조치를 취하는 것이 필요합니다:\n\n1. **자동재생 시도**: 우선적으로 일반적인 자동 재생을 진행하여 DPF 시스템이 정상적으로 작동하는지 확인합니다. 차량 속도는 약 60km/h로 유지하고, 20분 이상 주행하여 DPF 시스템이 스스로 soot를 제거하도록 합니다.\n\n2. **DPF 수동 제거 버튼 활용**: 만약 자동재생이 어려운 상태라면, DPF 수동 제거 버튼을 통해 운전자가 강제로 soot를 제거하는 방법을 시도해 봅니다.\n\n3. **정비소 방문**: 자동 재생이나 수동 제거 버튼을 통해서도 문제가 해결되지 않을 경우, 가까운 정비소를 방문하여 DPF 시스템을 점검 받는 것이 중요합니다. 이는 센서 불량이나 DPF 필터 문제 등 다른 원인을 파악하는 데 도움이 됩니다.\n\n4. **배기계통 점검**: 해당 현상이 지속 발견될 경우 만약을 대비하여 차량 자체의 배기계통 점검을 하는 것이 추천됩니다.\n\n**정상적인 경우**: 만약 soot 수치가 적절한 타이밍에 연소되어 없어지고 전체적으로 봤을 때 축적량이 40을 초과하는 구간이 존재하지 않는다면, 이는 정상적이며 문제 없습니다"
+warning  = "⚠️ 일정 수치 이상의 Soot 축적량이 2시간 이상 지속되는 현상을 보임. \nSoot 가 일정수치에 도달했을때 정상적으로 연소되지 않거나, 센서 불량으로 인해 Dosing 이 과하게 활성화됨. \nsoot 가 증가하는 과정에서 불규칙적인 증감 현상이 발생함. \n튐 또는 비선형 형태가 크게 나타남. \nSoot 의 축적량이 40을 초과하는 구간이 존재함."
 
-# 드롭다운 메뉴
-st.subheader("데이터 선택")
+def chart_generate(df, col_name):
+    return  st.line_chart(df, y=[col_name])
 
-options = [
-    "",
-    "2024-08-24",
-    "2024-09-26"
-]
-selected_text = st.selectbox("데이터를 선택하세요", options)
+def tts_engine(text):
+    cleaned_text = re.sub(r'[\x00-\x1F]+', '', text).replace("*", "")
+    tts = gTTS(text=cleaned_text, lang='ko')
+    mp3_fp = BytesIO()
+    tts.write_to_fp(mp3_fp)
+    # 오디오 스트림 재설정 후 재생
+    mp3_fp.seek(0)
+    # Base64 인코딩
+    audio_bytes = mp3_fp.read()
+    audio_base64 = base64.b64encode(audio_bytes).decode()
+    return audio_base64
 
-# 커스텀 CSS 삽입
-st.markdown("""
-    <style>
-    div.stButton > button {
-        background-color: transparent;
-        border: none;
-        color: black;
-        font-size: 30px;
-        padding: 0;
-    }
-    div.stButton > button:hover {
-        color: red;
-        cursor: pointer;
-    }
-    </style>
-""", unsafe_allow_html=True)
-df = pd.read_csv("https://huggingface.co/spaces/randmimc/mobility_condition/raw/main/chat_icon/text.csv", on_bad_lines='skip') 
+def none_temp():
+    st.info("먼저 데이터를 선택해 주세요")
 
-# 데이터 상태 및 조치 상황 표시 함수
-def display_status_and_action(text):
+def normal(df, col_name):
+    chart_generate(df, col_name)
+    st.success("✅ 차량 상태: 정상")
+    st.markdown("**조치 사항:** 추가 조치는 필요하지 않습니다.")
 
-    if text == "":
-        st.info("먼저 데이터를 선택해주세요.")
-    elif text == "2024-09-26":
-        st.line_chart(
-            df,
-            y=['ECU_DPFSoot']
-        )
-        st.success("✅ 차량 상태: 정상")
-        st.markdown("**조치 사항:** 추가 조치는 필요하지 않습니다.")
-    else:
-        st.line_chart(
-            df,
-            y=['ECU_DPFSoot']
-        )
-        text_input = "위의 증상들이 보일 경우, 다음과 같은 조치를 취하는 것이 필요합니다:\n\n1. **자동재생 시도**: 우선적으로 일반적인 자동 재생을 진행하여 DPF 시스템이 정상적으로 작동하는지 확인합니다. 차량 속도는 약 60km/h로 유지하고, 20분 이상 주행하여 DPF 시스템이 스스로 soot를 제거하도록 합니다.\n\n2. **DPF 수동 제거 버튼 활용**: 만약 자동재생이 어려운 상태라면, DPF 수동 제거 버튼을 통해 운전자가 강제로 soot를 제거하는 방법을 시도해 봅니다.\n\n3. **정비소 방문**: 자동 재생이나 수동 제거 버튼을 통해서도 문제가 해결되지 않을 경우, 가까운 정비소를 방문하여 DPF 시스템을 점검 받는 것이 중요합니다. 이는 센서 불량이나 DPF 필터 문제 등 다른 원인을 파악하는 데 도움이 됩니다.\n\n4. **배기계통 점검**: 해당 현상이 지속 발견될 경우 만약을 대비하여 차량 자체의 배기계통 점검을 하는 것이 추천됩니다.\n\n**정상적인 경우**: 만약 soot 수치가 적절한 타이밍에 연소되어 없어지고 전체적으로 봤을 때 축적량이 40을 초과하는 구간이 존재하지 않는다면, 이는 정상적이며 문제 없습니다"
-        warning  = "⚠️ 일정 수치 이상의 Soot 축적량이 2시간 이상 지속되는 현상을 보임. \nSoot 가 일정수치에 도달했을때 정상적으로 연소되지 않거나, 센서 불량으로 인해 Dosing 이 과하게 활성화됨. \nsoot 가 증가하는 과정에서 불규칙적인 증감 현상이 발생함. \n튐 또는 비선형 형태가 크게 나타남. \nSoot 의 축적량이 40을 초과하는 구간이 존재함."
-        st.warning(warning)
-        st.markdown(text_input)
+def anormal(df, col_name, warning, text_input, audio_base64):
+    # custom style
+    audio_html = f"""
+        <style>
+            .audio-container {{
+                display: flex;
+                align-items: center;
+                gap: 10px;
+                justify-content: flex-end; /* 오른쪽 정렬 */
+                margin-top: 10px;
+            }}
+            .audio-icon {{
+                font-size: 30px;
+                cursor: pointer;
+            }}
+            .audio-icon[title]:hover::after {{
+                content: attr(title);
+                position: absolute;
+                background: #333;
+                color: #fff;
+                padding: 5px 8px;
+                border-radius: 5px;
+                white-space: nowrap;
+                font-size: 12px;
+                top: -35px;
+                right: 0;
+                z-index: 999;
+            }}
+        </style>
+                    
+        <div class="audio-container">
+            <div class="audio-icon" title="위 내용을 음성으로 듣기">🔊</div>
+            <audio controls>
+                <source src="data:audio/mp3;base64,{audio_base64}" type="audio/mp3">
+                브라우저가 오디오 태그를 지원하지 않습니다.
+            </audio>
+        </div>
+    """
+    chart_generate(df, col_name)
+    st.warning(warning)
+    st.markdown(text_input)
+    st.markdown(audio_html, unsafe_allow_html=True)
 
-        text_input = warning + " " + text_input
-        cleaned_text = re.sub(r'[\x00-\x1F]+', '', text_input).replace("*", "")
-        tts = gTTS(text=cleaned_text, lang='ko')
-        mp3_fp = BytesIO()
-        tts.write_to_fp(mp3_fp)
+# 서비스 실행
+st.subheader("데이터 선택")
+selected_text = st.selectbox("데이터를 선택하세요", options)
+
+# st.markdown("""
+#    <style>
+#    div.stButton > button {
+#        background-color: transparent;
+#        border: none;
+#        color: black;
+#        font-size: 30px;
+#        padding: 0;
+#    }
+#    div.stButton > button:hover {
+#        color: red;
+#        cursor: pointer;
+#    }
+#    </style>
+# """, unsafe_allow_html=True)
 
-        # 오디오 스트림 재설정 후 재생
-        mp3_fp.seek(0)
-        
-        # Base64 인코딩
-        audio_bytes = mp3_fp.read()
-        audio_base64 = base64.b64encode(audio_bytes).decode()        
-        
-        # HTML 스타일 및 오디오 출력
-        audio_html = f"""
-            <style>
-                .audio-container {{
-                    display: flex;
-                    align-items: center;
-                    gap: 10px;
-                    justify-content: flex-end; /* 오른쪽 정렬 */
-                    margin-top: 10px;
-                }}
-                .audio-icon {{
-                    font-size: 30px;
-                    cursor: pointer;
-                }}
-                .audio-icon[title]:hover::after {{
-                    content: attr(title);
-                    position: absolute;
-                    background: #333;
-                    color: #fff;
-                    padding: 5px 8px;
-                    border-radius: 5px;
-                    white-space: nowrap;
-                    font-size: 12px;
-                    top: -35px;
-                    right: 0;
-                    z-index: 999;
-                }}
-            </style>
-                        
-            <div class="audio-container">
-                <div class="audio-icon" title="위 내용을 음성으로 듣기">🔊</div>
-                <audio controls>
-                    <source src="data:audio/mp3;base64,{audio_base64}" type="audio/mp3">
-                    브라우저가 오디오 태그를 지원하지 않습니다.
-                </audio>
-            </div>
-        """
-        
-        # HTML 렌더링
-        st.markdown(audio_html, unsafe_allow_html=True)
 
 if selected_text:
-    display_status_and_action(selected_text)
+    if selected_text == "":
+        none_temp()
+    elif text == "2024-09-26":
+        normal(df, "ECU_DPFSoot")
+    else:
+        audio_base64 = tts_engine(warning+" "+0text_input)
+        anormal(df, "ECU_DPFSoot", warning, text_input, audio_base64)
+