updated supabase files

src/backend/api_routes/chat_history_db_api.py

@@ -1,4 +1,4 @@
-from fastapi import APIRouter
+from fastapi import APIRouter,HTTPException,status
 from services.schemas import ChatHistoryRequest
 from services import supabase_service
 from utils import logger
@@ -16,5 +16,20 @@ def store_chat_history(chat_history : ChatHistoryRequest):
     except Exception as e:
         raise f"Failed to create {e}"
     
+@router.get('/get-history')
+def get_chat_history(conversation_id: str):
+    try:
+        chat_history = supabase_service.get_chat_history(conversation_id)
+        logger.info(f"Chat history retrieved successfully for conversation ID: {conversation_id}")
+        return chat_history
+    except Exception as e:
+        logger.error(f"Error retrieving chat history for ID {conversation_id}: {e}")
+        raise HTTPException(
+            status_code=status.HTTP_500_INTERNAL_SERVER_ERROR,
+            detail="Failed to retrieve chat history. Please try again later."
+        )
+
+
+    
 
 

src/backend/services/supabase_service.py

@@ -52,37 +52,26 @@ def store_chat_history(conversation_id, new_messages):
         logger.error(f"Error: {e}")
         return {"error": str(e)}
 
-def get_chat_history(date):
+def get_chat_history(conversation_id):
     try:
-        prefix = f"{date}/"
-        files = supabase.storage.from_(SUPABASE_BUCKET).list(prefix)
-        chat_history = []
-
-        for file in files:
-            file_path = file['name']
-            response = supabase.storage.from_(SUPABASE_BUCKET).download(file_path)
-            chat_data = json.loads(response)
-            chat_history.append(chat_data)
+        file_path = f"chat-history/{conversation_id}.json"
+        existing_data = supabase.storage.from_(SUPABASE_BUCKET).download(file_path)
+        
+        if existing_data:
+            chat_data = json.loads(existing_data.decode('utf-8'))
+            logger.info("Chat history retrieved successfully!")
+            return chat_data
+        else:
+            logger.warning("No chat history found for the given conversation ID.")
+            return {"message": "No chat history found."}
 
-        return chat_history
     except Exception as e:
         logger.error(f"Error retrieving chat history: {e}")
-        return []
+        return {"error": str(e)}
 
-def create_bucket_with_file():
-    bucket_name = "chat-history"
+def create_bucket_with_file(bucket_name:str):
     try:
         supabase.storage.create_bucket(bucket_name)
-        print(f"Bucket '{bucket_name}' created successfully.")
+        logger.info(f"Bucket '{bucket_name}' created successfully.")
     except Exception as e:
-        print(f"Error creating bucket: {e}")
-
-
-
-conversation_id = "12345"
-messages = [
-    {"role": "system", "content": "Hello Friend."},
-]
-
-
-# store_chat_history(conversation_id,messages)
+        logger.error(f"Error creating bucket: {e}")

src/frontend/app/common_functions.py

@@ -140,4 +140,32 @@ def store_chat_history_in_db(conversation_id, messages):
         response = requests.post(API_URL, json=payload)
         logger.info("Successfully added the chat in db")
     except Exception as e:
-        logger.info(f"Failed to add the chat in db {e}")
+        logger.info(f"Failed to add the chat in db {e}")
+
+def get_chat_history_from_db(conversation_id):
+    try:
+        API_URL = f"http://127.0.0.1:8000/chat-db/get-history"
+        response = requests.post(API_URL,params={"conversation_id": conversation_id})
+        response.raise_for_status()
+        logger.info(f"Successfully retrieved chat history for conversation ID: {conversation_id}")
+        return response.json()
+    except Exception as e:
+        logger.info(f"Failed to get the chat history")
+    return {"error": "Failed to retrieve chat history. Please try again later."}
+
+def display_chat_history(st):
+    conversation_id = st.session_state.get("conversation_id")
+    if not conversation_id:
+        st.warning("Conversation ID is missing. Please provide a valid ID.")
+        return
+
+    try:
+        chat_history = get_chat_history_from_db(conversation_id)
+        if chat_history and isinstance(chat_history, dict) and 'error' not in chat_history:
+            st.success("Chat history loaded successfully!")
+            if st.sidebar.button(f"Show History for {conversation_id}",key="show_history_button"):
+                st.json(chat_history)
+        else:
+            st.info("No chat history found for this conversation ID or the data format is incorrect.")
+    except Exception as e:
+        st.error(f"Error retrieving chat history: {e}")

src/frontend/app/pinecone_data_handler.py

@@ -1,5 +1,5 @@
 import requests
-from frontend.app import common_functions
+from app import common_functions
 import streamlit as st
 
 

src/frontend/home.py

@@ -1,5 +1,5 @@
 import streamlit as st
-from frontend.app import common_functions
+from app import common_functions
 from PIL import Image  
 
 def render_homepage():
@@ -7,6 +7,7 @@ def render_homepage():
 
     # Page Configuration
     common_functions.config_homepage(st)
+
     common_functions.set_page_title(st)
 
 
@@ -21,11 +22,6 @@ def render_homepage():
     )
 
     with Home:
-        # st.markdown("### 🌟 Getting Started")
-        # st.markdown("""
-        # - Select **Admin Portal** for system configurations and data management.
-        # - Go to **Knowledge Base Explorer** to explore and manage knowledge entries.
-        # """)
         st.markdown("""
         ### 👋 Welcome to the Yuvabe Care Companion AI!
         This platform offers comprehensive tools to support your healthcare journey. Use the tabs above to navigate:

src/frontend/pages/chatbot.py

@@ -29,12 +29,12 @@ def fetch_health_advice(conversation_history):
 if "conversation_history" not in st.session_state:
     st.session_state.conversation_history = initialize_conversation()
 
-# Display chat history
-for message in st.session_state.conversation_history [-NUMBER_OF_MESSAGES_TO_DISPLAY:]:
-    role = message["role"]
-    avatar_image = "src/frontend/images/page_icon.jpg" if role == "assistant" else "src/frontend/images/page_icon.jpg" if role == "user" else None
-    with st.chat_message(role, avatar=avatar_image):
-        st.write(message["content"])
+# # Display chat history
+# for message in st.session_state.conversation_history [-NUMBER_OF_MESSAGES_TO_DISPLAY:]:
+#     role = message["role"]
+#     avatar_image = "src/frontend/images/page_icon.jpg" if role == "assistant" else "src/frontend/images/page_icon.jpg" if role == "user" else None
+#     with st.chat_message(role, avatar=avatar_image):
+#         st.write(message["content"])
 
 # User Input
 user_input = st.chat_input("Ask your health-related question:")
@@ -60,6 +60,8 @@ if user_input:
     # 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)
+    common_functions.display_chat_history(st)
+
 
     # Display only the assistant's latest response
     with st.chat_message('assistant'):