diff --git a/main.py b/main.py
index 7f1e4af..8037dee 100644
--- a/main.py
+++ b/main.py
@@ -1,33 +1,84 @@
 import streamlit as st
 import requests
 
+# Initialize session state for prompt history
 if "history" not in st.session_state:
     st.session_state.history = []
 
 
+# Function to send POST request to a local server
 def send_post_request(prompt):
-    url = "http://localhost:19194/"
+    url = "http://localhost:19194/"  # Update this to your actual server URL
     payload = prompt.title()
     response = requests.post(url, json=payload)
-
-    return response.text
-
-
-st.title("Prompt History and Post Request App")
-
-prompt = st.text_input("Enter your prompt:")
-
-if st.button("Submit"):
-    if prompt:
-        st.session_state.history.append(prompt)
-        response = send_post_request(prompt)
-
-        st.success(f"Response: {response}")
+    json = response.json()
+    if "response" in json:
+        return json["response"]
     else:
-        st.error("Please enter a prompt")
+        return json
+
+
+# Custom CSS to fix the input box at the bottom of the page
+st.markdown(
+    """
+    <style>
+    .fixed-input-container {
+        position: fixed;
+        bottom: 0;
+        left: 0;
+        width: 100%;
+        background-color: #f9f9f9;
+        padding: 10px;
+        box-shadow: 0px -2px 10px rgba(0, 0, 0, 0.1);
+        z-index: 999;
+    }
+    .input-container input {
+        width: 80%;
+        padding: 10px;
+        font-size: 16px;
+        border-radius: 5px;
+        border: 1px solid #ccc;
+        margin-right: 10px;
+    }
+    .input-container button {
+        padding: 10px 20px;
+        font-size: 16px;
+        border: none;
+        background-color: #4CAF50;
+        color: white;
+        border-radius: 5px;
+        cursor: pointer;
+    }
+    .input-container button:hover {
+        background-color: #45a049;
+    }
+    .prompt-history {
+        margin-bottom: 120px; /* Space to ensure prompt history doesn't overlap input box */
+    }
+    </style>
+    """,
+    unsafe_allow_html=True
+)
+
+st.title('Ask maargdarshak')
+
+if 'messages' not in st.session_state:
+    st.session_state.messages = []
+
+for message in st.session_state.messages:
+    st.chat_message(message['role']).markdown(message['content'])
+
+prompt = st.chat_input("Ask your question here")
+
+if prompt:
+    st.chat_message('user').markdown(prompt)
+    st.session_state.messages.append({'role': 'user', 'content': prompt})
+
+    response = send_post_request(prompt)
+    print(response)
+    st.chat_message('bot').markdown(response)
+    st.session_state.messages.append({'role': 'bot', 'content': response})
+    # st.session_state.history.append((prompt, response))
+    # st.experimental_rerun()
+
 
-# Display history of prompts
-if st.session_state.history:
-    st.write("### Prompt History")
-    for i, p in enumerate(st.session_state.history, 1):
-        st.write(f"{i}. {p}")