Update app.py

app.py

@@ -6,10 +6,82 @@ from langchain.vectorstores import Chroma
 from langchain.embeddings import HuggingFaceEmbeddings
 from langchain_core.prompts import ChatPromptTemplate, PromptTemplate
 from langchain_groq import ChatGroq
-from langchain.chains import RetrievalQA
+from langchain.chains import RetrievalQA, LLMChain
 from langchain.document_loaders import PyPDFLoader
 from langchain.text_splitter import RecursiveCharacterTextSplitter
-from langchain.chains import LLMChain
+
+# Set page configuration with wide layout
+st.set_page_config(
+    page_title="Dr. Radha: The Agro-Homeopath",
+    page_icon="🌿",
+    layout="wide",
+    initial_sidebar_state="collapsed"
+)
+
+# Enhanced CSS styling
+st.markdown("""
+    <style>
+    /* Main container styling */
+    .stApp {
+        background-color: #1B4D3E;
+    }
+    
+    /* Chat message styling */
+    .stChatMessage {
+        background-color: rgba(255, 255, 255, 0.1);
+        border-radius: 15px;
+        padding: 15px;
+        margin: 10px 0;
+        border: 1px solid rgba(255, 255, 255, 0.2);
+    }
+    
+    /* Input field styling */
+    .stChatInput {
+        border-radius: 20px !important;
+        border: 2px solid rgba(255, 255, 255, 0.2) !important;
+        background-color: rgba(255, 255, 255, 0.05) !important;
+    }
+    
+    /* Text colors */
+    .stMarkdown, .stText, .stTitle, .stHeader, .stSubheader {
+        color: white !important;
+    }
+    
+    /* Button styling */
+    .stButton > button {
+        background-color: #FFD700 !important;
+        color: #1B4D3E !important;
+        font-weight: bold;
+        border-radius: 25px;
+        padding: 10px 25px;
+        border: none;
+        transition: all 0.3s ease;
+    }
+    
+    .stButton > button:hover {
+        transform: scale(1.05);
+        box-shadow: 0 5px 15px rgba(0,0,0,0.2);
+    }
+    
+    /* Center alignment */
+    .css-10trblm, .css-1a7jz76 {
+        text-align: center !important;
+    }
+    </style>
+""", unsafe_allow_html=True)
+
+# Initialize session state for chat history
+if "messages" not in st.session_state:
+    st.session_state.messages = []
+    st.session_state.messages.append({
+        "role": "assistant",
+        "content": "👋 Hello! I'm Dr. Radha, your AI-powered Organic Farming Consultant. How can I assist you today?"
+    })
+
+# Your existing initialization code here
+PERSISTENT_DIR = "vector_db"
+# [Keep all your existing functions and variable definitions]
+
 
 # Set persistent storage path
 PERSISTENT_DIR = "vector_db"
@@ -58,53 +130,6 @@ api_key1 = os.getenv("api_key")
 start_time = time.time()
 st.set_page_config(page_title="Dr. Radha: The Agro-Homeopath", page_icon="🚀", layout="wide")
 
-# CSS for dark green background and white text
-st.markdown("""
-<style>
-/* Set only background color for app */
-.stApp {
-    background-color: #1B4D3E !important;
-}
-
-/* Style all text elements in white */
-.stMarkdown, .stText, .stTitle, .stHeader, .stSubheader, 
-.stTextInput label, .stSelectbox label, .st-emotion-cache-10trblm,
-.st-emotion-cache-1a7jz76, .st-emotion-cache-1629p8f,
-[data-testid="stTitle"], [data-testid="stSubheader"] {
-    color: white !important;
-}
-
-/* Additional specific selectors for title and subheader */
-h1, h2, h3 {
-    color: white !important;
-}
-
-/* Keep button text black */
-.stButton > button {
-    color: black !important;
-    background-color: yellow !important;
-}
-</style>
-""", unsafe_allow_html=True)
-
-# Add custom CSS for centering
-st.markdown("""
-<style>
-/* Center align title and header */
-.css-10trblm, .css-1a7jz76 {
-    text-align: center !important;
-    color: white !important;
-}
-
-/* Center align markdown text */
-.center-text {
-    text-align: center !important;
-    margin: 0 auto;
-    display: block;
-}
-</style>
-""", unsafe_allow_html=True)
-
 # Title and subheader
 st.title("🌿 Dr. Radha: AI-Powered Organic Farming Consultant")
 st.subheader("Specializing in Agro-Homeopathy | Free Consultation")
@@ -235,53 +260,51 @@ Answer:"""
 fallback_prompt = PromptTemplate(template=fallback_template, input_variables=["question"])
 fallback_chain = LLMChain(llm=llm, prompt=fallback_prompt)
 
+# Replace your existing chat container and form section with this:
 chat_container = st.container()
+with chat_container:
+    # Display chat history
+    for message in st.session_state.messages:
+        with st.chat_message(message["role"], avatar="👤" if message["role"] == "user" else "🌿"):
+            st.markdown(message["content"])
 
 with st.form(key='query_form', clear_on_submit=True):
-    query = st.text_input("Ask your question:", value="", help=None)
-    st.markdown(
-        """
-        <style>
-        .stTextInput label {
-            color: white !important;
-        }
-        .stButton > button {
-        color: black !important;
-        background-color: yellow !important;
-        font-weight: bold;
-        }
-        </style>
-        """, 
-        unsafe_allow_html=True
+    query = st.text_input(
+        "Ask your question:",
+        placeholder="Describe your plant issue here...",
+        label_visibility="collapsed"
     )
-    submit_button = st.form_submit_button(label='Submit')
-end_time = time.time()
-#print(f"Setting up retrieval chain took {end_time - start_time:.4f} seconds")
-start_time = time.time()
+    submit_button = st.form_submit_button(label='Send Message 📤')
 
 if submit_button and query:
-    with st.spinner("Generating response..."):
-        result = qa({"query": query})  
-        if result['result'].strip() == "":
-            # If no result from PDF, use fallback chain
-            fallback_result = fallback_chain.run(query)
-            response = fallback_result
-        else:
-            response = result['result']
-
-    col1, col2 = st.columns([1, 10])
-    with col1:
-        st.image(human_image, width=80)
-    with col2:
-        st.markdown(f"{query}")
-    col1, col2 = st.columns([1, 10])
-    with col1:
-        st.image(robot_image, width=80)
-    with col2:
-        st.markdown(f"{response}")
+    # Add user message to history
+    st.session_state.messages.append({"role": "user", "content": query})
+    
+    # Show user message
+    with st.chat_message("user", avatar="👤"):
+        st.markdown(query)
+    
+    # Show typing indicator while generating response
+    with st.chat_message("assistant", avatar="🌿"):
+        with st.status("Analyzing your query...", expanded=True):
+            st.write("🔍 Retrieving relevant information...")
+            st.write("📝 Generating personalized response...")
+            
+            # Generate response
+            result = qa({"query": query})
+            if result['result'].strip() == "":
+                response = fallback_chain.run(query)
+            else:
+                response = result['result']
+            
+            # Display final response
+            st.markdown(response)
     
-    st.markdown("---")
+    # Add assistant response to history
+    st.session_state.messages.append({"role": "assistant", "content": response})
     
+    # Clear the form
     st.session_state["query"] = ""
-
-end_time = time.time()
+    
+    # Rerun to update chat history
+    st.rerun()