added chatbot feature

app.py

@@ -1,5 +1,126 @@
 import streamlit as st
+from langchain.prompts import PromptTemplate
+from langchain.llms import CTransformers
+from src.helper import download_hf_embeddings, text_split, download_hf_model
+from langchain.vectorstores import Pinecone
+import os
+from dotenv import load_dotenv
+from src.prompt import prompt_template
+from langchain.chains import RetrievalQA
+import time
 
-x = st.slider('Select a value')
-st.write(x, 'squared is', x * x)
+# Load environment variables
+load_dotenv()
+PINECONE_API_KEY = os.environ.get('PINECONE_API_KEY')
 
+# Set page configuration
+st.set_page_config(page_title="Medical Chatbot", page_icon="🏥", layout="wide")
+
+# Custom CSS for styling
+st.markdown("""
+<style>
+    .stApp {
+        background-color: #f0f8ff;
+    }
+    .stButton>button {
+        background-color: #4CAF50;
+        color: white;
+        border-radius: 20px;
+        border: none;
+        padding: 10px 20px;
+        transition: all 0.3s ease;
+    }
+    .stButton>button:hover {
+        background-color: #45a049;
+        transform: scale(1.05);
+    }
+    .footer {
+        position: fixed;
+        left: 0;
+        bottom: 0;
+        width: 100%;
+        background-color: #333;
+        color: white;
+        text-align: center;
+        padding: 10px 0;
+    }
+    .social-icons a {
+        color: white;
+        margin: 0 10px;
+        font-size: 24px;
+    }
+</style>
+""", unsafe_allow_html=True)
+
+# Initialize session state for chat history
+if 'chat_history' not in st.session_state:
+    st.session_state.chat_history = []
+
+# Header
+st.title("🏥 Medical Chatbot")
+
+# Initialize the chatbot components
+@st.cache_resource
+def initialize_chatbot():
+    embeddings = download_hf_embeddings()
+    model_name_or_path = "TheBloke/Llama-2-7B-Chat-GGML"
+    model_basename = "llama-2-7b-chat.ggmlv3.q4_0.bin"
+    model_path = download_hf_model(model_name_or_path, model_basename)
+    llm = CTransformers(model=model_path,
+                        model_type="llama",
+                        config={'max_new_tokens': 512,
+                                'temperature': 0.8})
+    PROMPT = PromptTemplate(template=prompt_template, input_variables=["context", "question"])
+    chain_type_kwargs = {"prompt": PROMPT}
+    docsearch = LangchainPinecone(index, embeddings.embed_query, "text")
+    qa = RetrievalQA.from_chain_type(
+        llm=llm,
+        chain_type="stuff",
+        retriever=docsearch.as_retriever(search_kwargs={'k': 2}),
+        return_source_documents=True,
+        chain_type_kwargs=chain_type_kwargs)
+    return qa
+
+qa = initialize_chatbot()
+
+# Chat interface
+user_input = st.text_input("Ask your medical question:")
+if st.button("Send", key="send"):
+    if user_input:
+        with st.spinner("Thinking..."):
+            result = qa({"query": user_input})
+            response = result["result"]
+        st.session_state.chat_history.append(("You", user_input))
+        st.session_state.chat_history.append(("Bot", response))
+
+# Display chat history
+st.subheader("Chat History")
+for role, message in st.session_state.chat_history:
+    if role == "You":
+        st.markdown(f"**You:** {message}")
+    else:
+        st.markdown(f"**Bot:** {message}")
+
+# Animated loading for visual appeal
+def load_animation():
+    with st.empty():
+        for i in range(3):
+            for j in ["⋅", "⋅⋅", "⋅⋅⋅", "⋅⋅⋅⋅"]:
+                st.write(f"Loading{j}")
+                time.sleep(0.2)
+            st.write("")
+
+# Footer with social links
+st.markdown("""
+<div class="footer">
+    <div class="social-icons">
+        <a href="https://github.com/yourusername" target="_blank"><i class="fab fa-github"></i></a>
+        <a href="https://linkedin.com/in/yourusername" target="_blank"><i class="fab fa-linkedin"></i></a>
+        <a href="https://yourwebsite.com" target="_blank"><i class="fas fa-globe"></i></a>
+    </div>
+    <p>© 2024 Medical Chatbot. All rights reserved.</p>
+</div>
+""", unsafe_allow_html=True)
+
+# Load Font Awesome for icons
+st.markdown('<link rel="stylesheet" href="https://cdnjs.cloudflare.com/ajax/libs/font-awesome/5.15.1/css/all.min.css">', unsafe_allow_html=True)