Update app.py

app.py

@@ -8,14 +8,13 @@ from arabic_support import support_arabic_text
 from pathlib import Path
 import json
 from PIL import Image
-from config.config import *
-
+from config.conf import *
 
 st.set_page_config(
-    page_title="AI-Medical Questionnaire",
-    page_icon="💊",
-    layout="wide",
-    initial_sidebar_state="expanded",
+        page_title="AI-Medical Questionnaire",
+        page_icon="💊",
+        layout="wide",
+        initial_sidebar_state="expanded",
 )
 
 # مسیر فایل فعلی
@@ -37,8 +36,8 @@ with open(css_file_path, "r", encoding="utf-8") as file:
 
 # اعمال CSS
 st.markdown(f"<style>{css_code}</style>", unsafe_allow_html=True)
-def go_to_level(selected):
-    state.level = levels()
+# def go_to_level(selected):
+#     state.level = levels()
 support_arabic_text(all=True)
 state = st.session_state
 st.title("AI-Medical Questionnaire")
@@ -46,6 +45,13 @@ side = st.sidebar
 
 if "messages" not in state:
     state["messages"] = []
+    start = """ سلام من دستیار شما هستم. برای بررسی و تصمیم گیری وضعیت بیماران مرحله به مرحله بررسی وضعیت بیمار را با کمک من انجام دهید. برای پیشنهاد می کنم 
+    اختلالات حرکتی را بررسی کنید. آیا بیمار دارای چنین اختلالی است؟)"""
+    state.messages.append({"role": "assistant", "content": start})
+    # st.chat_message("user", avatar=str(assets_dir / "user.png")).markdown(
+    # state["messages"] = []
+if "result" not in state:
+    state.result = ""
 if "level" not in state:
     state.level = "L01"
     state.next = ""
@@ -67,75 +73,58 @@ question_text = question_class.get_question_text(state.level)
 with side:
     img = Image.open(assets_dir / "logo.png")
     st.image(
-        img,
-        width=250,
-        caption="جهت بازنشانی از دکمه ریست استفاده کنید",
+            img,
+            width=250,
+            caption="جهت بازنشانی از دکمه ریست استفاده کنید",
     )
-    colm1, colm2 = st.columns(2)
-    with colm1:
-
-        if st.button("change"):
-            prv = get_response(state.previous)
-            next_level = option_class.get_next_level(prv, state.previous)
-            feedback = option_class.get_feedback(prompt, state.previous)
-    
-            if next_level != "L98" and next_level != "L99":
-                my_bar = st.progress(0, text=feedback)
-                for percent_complete in range(100):
-                    time.sleep(0.01)
-                    my_bar.progress(percent_complete + 1, text=feedback)
-                time.sleep(1)
-                my_bar.empty()
-                state.level = state.previous
-                state.level = next_level
-                
-                resp = question_class.get_question_text(state.level)[0]
-                with st.chat_message("assistant"):
-                    response = st.write_stream(response_generator(resp))
-                    state.messages.append({"role": "assistant", "content": f"checkpoint:  {state.previous}"})
-                    state.messages.append({"role": "assistant", "content": response})
-        with colm2:
-            if st.button("reset"):
-                            
-                del state["messages"]
-                st.rerun()
 
+    if st.button("reset"):
+
+        del state["messages"]
+        st.rerun()
 
 def response_generator(resp):
-    # resp = question_class.get_question_text(state.level)[0]
+    yield f"checkpoint: {state.previous}"
+    yield "\n\n"
     for word in resp.split():
         character = ""
         for char in word:
             yield character + char
-            time.sleep(0.02)
+            time.sleep(0.03)
         yield " "
-        time.sleep(0.09)
+        time.sleep(0.05)
 
-if "start" not in state:
-    state.start = """ سلام من دستیار شما هستم. برای بررسی و تصمیم گیری وضعیت بیماران مرحله به مرحله بررسی وضعیت بیمار را با کمک من انجام دهید. برای پیشنهاد می کنم 
-    اختلالات حرکتی را بررسی کنید. آیا بیمار دارای چنین اختلالی است؟)"""
 
-st.chat_message("assistant", avatar=str(assets_dir / "robot_icon.png")).markdown(
-    state.start
-)
 for messages in state.messages:
-    with st.chat_message(messages["role"]):
+
+    if messages["role"] == "assistant":
+        avatar = str(assets_dir / "robot_icon.png")
+    elif messages["role"] == "user":
+        avatar = str(assets_dir / "user.png")
+    else:
+        avatar = None  # Default
+
+    with st.chat_message(messages["role"], avatar=avatar):
         st.markdown(messages["content"])
 
+
 prompt = st.chat_input("""نوشتن پیام""")
 
 if prompt:
-    if prompt == "L99":
+    answer = prompt
+    answer = get_response(answer)
+    if answer == "L99":
+        state.result = "L99"
+
+        feedback = option_class.get_feedback("نتیجه", state.result)
         st.switch_page("pages/result.py")
-    prompt = is_valid_input(prompt)
     st.chat_message("user", avatar=str(assets_dir / "user.png")).markdown(prompt)
     state.messages.append({"role": "user", "content": prompt})
 
-    if prompt == "بله" or prompt == "خیر":
+    if answer == "بله" or answer == "خیر":
 
-        prompt = get_response(prompt)
-        next_level = option_class.get_next_level(prompt, state.level)
-        feedback = option_class.get_feedback(prompt, state.level)
+        next_level = option_class.get_next_level(answer, state.level)
+        feedback = option_class.get_feedback(answer, state.level)
 
         if next_level != "L98" and next_level != "L99":
             my_bar = st.progress(0, text=feedback)
@@ -146,32 +135,35 @@ if prompt:
             my_bar.empty()
             state.previous = state.level
             state.level = next_level
-            
+
             resp = question_class.get_question_text(state.level)[0]
             with st.chat_message("assistant"):
                 response = st.write_stream(response_generator(resp))
-                state.messages.append({"role": "assistant", "content": f"checkpoint:  {state.previous}"})
                 state.messages.append({"role": "assistant", "content": response})
-        
+
         elif next_level == "L98":
+            state.previous = state.level
+            state.level = next_level
+
+            state.result = feedback
             st.switch_page("pages/result.py")
         elif next_level == "L99":
+            state.result = feedback
             st.switch_page("pages/result.py")
-    else:
-        prompt = get_response(prompt)
 
-        if prompt[0] == "L":
-            feedback = "تغییر بخش مورد بررسی, تا چند لحظه دیگر"
-            state.previous = state.level
-            next_level = prompt
-            state.level = next_level
-            resp = question_class.get_question_text(state.level)[0]
-            with st.chat_message("assistant"):
-                response = st.write_stream(response_generator(resp))
-                state.messages.append({"role": "assistant", "content": f"checkpoint:  {state.previous}"})
-                state.messages.append({"role": "assistant", "content": response})
+    elif answer[0] == "L":
 
-        else:
-            with st.chat_message("assistant"):
-                response = st.write_stream(response_generator(prompt))
-                state.messages.append({"role": "assistant", "content": response})
+        feedback = "تغییر بخش مورد بررسی, تا چند لحظه دیگر"
+        state.previous = state.level
+        next_level = answer
+        state.level = next_level
+        resp = question_class.get_question_text(state.level)[0]
+        with st.chat_message("assistant"):
+            response = st.write_stream(response_generator(resp))
+            state.messages.append({"role": "assistant", "content": response})
+
+    else:
+        with st.chat_message("assistant"):
+            answer = unknown()
+            response = st.write_stream(response_generator(answer))
+            state.messages.append({"role": "assistant", "content": response})