update app.py

app.py

@@ -6,16 +6,21 @@ from peft import PeftModel, PeftConfig
 
 # Hugging Face login
 token = os.environ.get("token")
+if not token:
+    raise ValueError("Token not found. Please set the 'token' environment variable.")
 login(token)
-print("login is succesful")
-max_length=512
+print("Login is successful")
 
 # Model and tokenizer setup
 MODEL_NAME = "google/flan-t5-base"
-tokenizer = T5Tokenizer.from_pretrained(MODEL_NAME, use_auth_token=token)
-config = PeftConfig.from_pretrained("Komal-patra/results")
-base_model = AutoModelForSeq2SeqLM.from_pretrained(MODEL_NAME)
-model = PeftModel.from_pretrained(base_model, "Komal-patra/results")
+try:
+    tokenizer = T5Tokenizer.from_pretrained(MODEL_NAME, use_auth_token=token)
+    config = PeftConfig.from_pretrained("Komal-patra/results")
+    base_model = AutoModelForSeq2SeqLM.from_pretrained(MODEL_NAME)
+    model = PeftModel.from_pretrained(base_model, "Komal-patra/results")
+except Exception as e:
+    print(f"Error loading model: {e}")
+    raise
 
 # Text generation function
 def generate_text(prompt, max_length=512):
@@ -30,71 +35,66 @@ def generate_text(prompt, max_length=512):
     return generated_text
 
 # Custom CSS for the UI
-background_image_path = 'https://www.shlegal-technology.com/sites/default/files/insight/ExploringTheLegislativeBackgroundBANNER.jpg'
-custom_css = f"""
-.message.pending {{
+custom_css = """
+.message.pending {
     background: #A8C4D6;
-}}
+}
 /* Response message */
-.message.bot.svelte-1s78gfg.message-bubble-border {{
+.message.bot.svelte-1s78gfg.message-bubble-border {
     border-color: #266B99;
-}}
+}
 /* User message */
-.message.user.svelte-1s78gfg.message-bubble-border {{
+.message.user.svelte-1s78gfg.message-bubble-border {
     background: #9DDDF9;
     border-color: #9DDDF9;
-}}   
+}   
 /* For both user and response message as per the document */
-span.md.svelte-8tpqd2.chatbot.prose p {{
+span.md.svelte-8tpqd2.chatbot.prose p {
     color: #266B99;
-}}
+}
 /* Chatbot container */
-.gradio-container {{
-    background: #1c1c1c; /* Dark background */
+.gradio-container {
+    background: #84d5f7; /* Light blue background */
     color: white; /* Light text color */
-    background-image: url('{background_image_path}'); /* Add background image */
-    background-size: cover; /* Cover the entire container */
-    background-position: center; /* Center the image */
-    background-repeat: no-repeat; /* Do not repeat the image */
-}}
+}
 /* RED (Hex: #DB1616) for action buttons and links only */
-.clear-btn {{
+.clear-btn {
     background: #DB1616;
     color: white;
-}}
+}
 /* Primary colors are set to be used for all sorts */
-.submit-btn {{
+.submit-btn {
     background: #266B99;
     color: white;
-}}
+}
 /* Add icons to messages */
-.message.user.svelte-1s78gfg {{
+.message.user.svelte-1s78gfg {
     display: flex;
     align-items: center;
-}}
-.message.user.svelte-1s78gfg:before {{
-    content: url('file=Komal-patra/EU_AI_ACT/user icon.jpeg');
+}
+.message.user.svelte-1s78gfg:before {
+    content: url('file=Komal-patra/EU_AI_ACT/user_icon.jpeg');
     margin-right: 8px;
-}}
-.message.bot.svelte-1s78gfg {{
+}
+.message.bot.svelte-1s78gfg {
     display: flex;
     align-items: center;
-}}
-.message.bot.svelte-1s78gfg:before {{
-    content: url('file=Komal-patra/EU_AI_ACT/orcawise image.png');
+}
+.message.bot.svelte-1s78gfg:before {
+    content: url('file=Komal-patra/EU_AI_ACT/orcawise_image.png');
     margin-right: 8px;
-}}
+}
 /* Enable scrolling for the chatbot messages */
-.chatbot .messages {{
+.chatbot .messages {
     max-height: 500px;  /* Adjust as needed */
     overflow-y: auto;
-}}
+}
 """
 
 # Gradio interface setup
 with gr.Blocks(css=custom_css) as demo:
     chatbot = gr.Chatbot()
-    msg = gr.Textbox(placeholder="Ask your question...", show_label=False)  # Add placeholder text
+    msg = gr.Textbox(placeholder="Ask your question...", show_label=False)
     submit_button = gr.Button("Submit", elem_classes="submit-btn")
     clear = gr.Button("Clear", elem_classes="clear-btn")