Update app.py

app.py

@@ -48,40 +48,15 @@ rag_chain = (
 )
 import gradio as gr
 
-def rag_memory_stream(text):
+
+def rag_memory_stream(message, history):
     partial_text = ""
-    for new_text in rag_chain.stream(text):
+    for new_text in rag_chain.stream(message):
         partial_text += new_text
         yield partial_text
 
 examples = ['I feel dizzy', 'what is the possible sickness for fatigue']
 
-with open('index.html', 'r') as file:
-    custom_html = file.read()
-css = """body {
-    font-family: Arial, sans-serif;
-    background-color: #f0f0f0;
-    display: flex;
-    justify-content: center;
-    align-items: center;
-    height: 100vh;
-    margin: 0;
-}
-
-.chat-container {
-    background-color: white;
-    border-radius: 10px;
-    box-shadow: 0 4px 8px rgba(0,0,0,0.1);
-    padding: 20px;
-    width: 100%;
-    max-width: 600px;
-}
-
-h1 {
-    text-align: center;
-    color: #333;
-}
-"""
 
 title = "MediGuide ChatBot"
 demo = gr.Interface(
@@ -90,8 +65,9 @@ demo = gr.Interface(
     inputs="text",
     outputs="text",
     examples=examples,
-    css=css,
     allow_flagging="never",
+    fill_height=True,
+    theme="glass",
 )