Update app.py

app.py

@@ -1,14 +1,24 @@
-# main.py
-
 from openai import OpenAI
-import gradio as gr
 from gradio.chat_interface import ChatInterface
 import os
 from dotenv import load_dotenv
+from gradio.themes.soft import Soft
+from gradio.themes import colors
+from gradio.components import Textbox
+from gradio.components import Chatbot
+from RAG_class import RAG_1177
 load_dotenv()
 
-from RAG_class import RAG_1177
+textbox = Textbox(placeholder="Skriv ditt fråga här...",scale=4)
+chatbot = Chatbot(placeholder="Varsågod och ställ din fråga nedanför 😊",scale=3, height=250, show_copy_button=True, label="1177 chatbot")
+
+new_primary_color = colors.red
+new_secondary_color = colors.red
 
+my_custom_theme = Soft(
+    primary_hue=new_primary_color,
+    secondary_hue=new_secondary_color
+)
 
 # System prompt
 system_prompt = """
@@ -31,7 +41,8 @@ examples = [
         "Vad är symptomen för covid-19?",
         "Vilka läkemedel finns det för smärtlindring?",
         "Hur kan jag bäst lindra mina symtom på pollenallergi?",
-        "Vad kan jag göra för att förbättra min sömn?"
+        "Vad kan jag göra för att förbättra min sömn?",
+        "vad kan jag göra om mitt barn får magsjuka?"  
     ]
 
 
@@ -91,8 +102,11 @@ def predict(message, history):
     yield partial_message
 
 def main():
-  
-    ChatInterface(predict, title="Välkommen till 1177 AI-chatbot! 🔍🤒", theme="soft",cache_examples=False, examples=examples, description=description).launch(share=False)
-    
+
+    ChatInterface(predict, textbox=textbox, chatbot=chatbot, title="Välkommen till 1177 AI-chatbot! 🔍🤒", theme=my_custom_theme,submit_btn="Skicka",
+                  retry_btn="🔄Försök igen", undo_btn="↩️ Ångra", clear_btn="🗑️ Rensa",
+                   examples=examples,  cache_examples=False, description=description).launch(share=True)
+
 if __name__ == "__main__":
     main()
+