Spaces:
Sleeping
Sleeping
| import streamlit as st | |
| import requests | |
| from frontend.app import common_functions | |
| from datetime import datetime | |
| API_URL = "http://localhost:8000/chat/get-health-advice/" | |
| NUMBER_OF_MESSAGES_TO_DISPLAY = 20 | |
| common_functions.config_homepage() | |
| common_functions.set_page_title() | |
| # common_functions.set_bg_image("src/frontend/images/health_care_baner_2.jpg") | |
| # Initialize conversation history | |
| def initialize_conversation(): | |
| assistant_message = ("Hello! I am your Yuvabe Care Companion AI, here to assist you with general medicine queries. " | |
| "How can I help you today?") | |
| return [{"role": "assistant", "content": assistant_message}] | |
| # Function to fetch advice from the API | |
| def fetch_health_advice(conversation_history): | |
| try: | |
| response = requests.post( | |
| API_URL, | |
| json={"conversation_history": conversation_history} | |
| ) | |
| response.raise_for_status() | |
| return response.json().get("reply", "I couldn't process your request at the moment.") | |
| except requests.exceptions.RequestException as e: | |
| st.error(f"API Connection Error: {e}") | |
| return "I'm currently unable to respond. Please try again later." | |
| conversation_ids = common_functions.get_bucket_items() | |
| if conversation_ids: | |
| for conversation_id in conversation_ids[-3:]: | |
| common_functions.display_chat_history(conversation_id) | |
| def render_chatbot(): | |
| if "conversation_history" not in st.session_state: | |
| st.session_state.conversation_history = [] | |
| if 'conversation_id' not in st.session_state: | |
| st.session_state.conversation_id = datetime.now().strftime("%Y-%m-%d") | |
| # Display chat history | |
| for message in st.session_state.conversation_history [-NUMBER_OF_MESSAGES_TO_DISPLAY:]: | |
| role = message["role"] | |
| avatar_image = "src/frontend/images/chat_doctor_logo.png" if role == "assistant" else "src/frontend/images/healthy.png" if role == "user" else None | |
| with st.chat_message(role, avatar=avatar_image): | |
| common_functions.display_message_box(role,message['content']) | |
| # User Input | |
| user_input = st.chat_input("Ask your health-related question:") | |
| if 'system_message' not in st.session_state: | |
| system_message = ("Hello! I am your Yuvabe Care Companion AI, here to assist you with general medicine queries. " | |
| "How can I help you today?") | |
| st.session_state.system_message = system_message | |
| with st.chat_message('ai'): | |
| common_functions.typewriter_effect(st.session_state.system_message,speed=0) | |
| if user_input: | |
| # Display user's input | |
| user_avatar_image = "src/frontend/images/healthy.png" | |
| with st.chat_message('user',avatar=user_avatar_image): | |
| common_functions.typewriter_effect(user_input) | |
| # Append user input to session history | |
| st.session_state.conversation_history.append({"role": "user", "content": user_input}) | |
| # Fetch assistant response | |
| assistant_reply = fetch_health_advice(st.session_state.conversation_history) | |
| # Append assistant's reply to conversation history first | |
| st.session_state.conversation_history.append({"role": "assistant", "content": assistant_reply}) | |
| common_functions.store_chat_history_in_db(st.session_state.conversation_id,st.session_state.conversation_history) | |
| # Display only the assistant's latest response | |
| doctor_avatar_image = "src/frontend/images/chat_doctor_logo.png" | |
| with st.chat_message('assistant',avatar=doctor_avatar_image): | |
| common_functions.typewriter_effect(assistant_reply) | |
| if __name__ == "__main__": | |
| render_chatbot() | |