Update app.py

app.py

@@ -47,29 +47,45 @@ rag_chain = (
 )
 import gradio as gr
 
-def rag_memory_stream(message, history):
-    partial_text = ""
-    for new_text in rag_chain.stream(message):
-        partial_text += new_text
-        yield partial_text
-
-greetingsmessage = """Hello! Welcome to MediGuide ChatBot. I'm here to provide you with quick and accurate information on medical drugs. 
-Whether you need details on usage, side effects , etc feel free to ask. Let's enhance patient care together!"""
-
-
-title = "MediGuide ChatBot"
-
-demo = gr.Interface(
-     
-    title=title,
-    fn=rag_memory_stream,
-    inputs='text',
-    outputs='text',
-    allow_flagging="never",
-    fill_height=True,
-    theme=gr.themes.Soft(),
-)
-
+# Function to handle prompt and query the RAG chain
+def handle_prompt(message, history):
+    try:
+        # Stream output
+        out=""
+        for chunk in rag_chain.stream(message):
+            out += chunk
+            yield out
+    except:
+        raise gr.Error("Requests rate limit exceeded")
+
+
+if __name__=="__main__":
+
+    # Predefined messages and examples
+    description = "AI powered assistant which answers any question related to the [CAMELS simulations](https://www.camel-simulations.org/)."
+    greetingsmessage = "Hi, I'm the CAMELS DocBot, I'm here to assist you with any question related to the CAMELS simulations."
+    example_questions = [
+                        "How can I read a halo file?",
+                        "Which simulation suites are included in CAMELS?",
+                        "Which are the largest volumes in CAMELS simulations?",
+                        "Write a complete snippet of code getting the power spectrum of a simulation"
+                        ]
+
+    # Define customized Gradio chatbot
+    chatbot = gr.Chatbot([{"role":"assistant", "content":greetingsmessage}],
+                        type="messages",
+                        avatar_images=["ims/userpic.png","ims/camelslogo.jpg"],
+                        height="60vh")
+
+    # Define Gradio interface
+    demo = gr.ChatInterface(handle_prompt,
+                            type="messages",
+                            title="CAMELS DocBot",
+                            fill_height=True,
+                            examples=example_questions,
+                            theme=gr.themes.Soft(),
+                            description=description,
+                            #cache_examples=False,
+                            chatbot=chatbot)
 
-if __name__ == "__main__":
     demo.launch()