Update

app.py

@@ -6,11 +6,9 @@ from peft import PeftModel, PeftConfig
 
 # Hugging Face login
 token = os.environ.get("token")
-if token:
-    login(token)
-    print("Login is successful")
-else:
-    print("Token not found. Please set your token in the environment variables.")
+login(token)
+print("login is succesful")
+max_length=512
 
 # Model and tokenizer setup
 MODEL_NAME = "google/flan-t5-base"
@@ -20,7 +18,7 @@ base_model = AutoModelForSeq2SeqLM.from_pretrained(MODEL_NAME)
 model = PeftModel.from_pretrained(base_model, "Komal-patra/results")
 
 # Text generation function
-def generate_text(prompt, max_length=150):
+def generate_text(prompt, max_length=512):
     inputs = tokenizer(prompt, return_tensors="pt")
     outputs = model.generate(
         input_ids=inputs["input_ids"],
@@ -34,34 +32,31 @@ def generate_text(prompt, max_length=150):
 # Custom CSS for the UI
 background_image_path = 'https://www.shlegal-technology.com/sites/default/files/insight/ExploringTheLegislativeBackgroundBANNER.jpg'
 custom_css = f"""
-/* Chatbot container */
-.gradio-container {{
-    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 */
-    background-color: rgba(0, 0, 0, 0.5); /* Semi-transparent dark background */
-}}
-
-/* Transparent message bubbles */
 .message.pending {{
-    background: rgba(168, 196, 214, 0.1);
+    background: #A8C4D6;
 }}
+/* Response message */
 .message.bot.svelte-1s78gfg.message-bubble-border {{
-    border-color: rgba(38, 107, 153, 0.1);
-    background: rgba(255, 255, 255, 0.1);
+    border-color: #266B99;
 }}
+/* User message */
 .message.user.svelte-1s78gfg.message-bubble-border {{
-    background: rgba(157, 221, 249, 0.1);
-    border-color: rgba(157, 221, 249, 0.1);
+    background: #9DDDF9;
+    border-color: #9DDDF9;
 }}   
-
 /* For both user and response message as per the document */
 span.md.svelte-8tpqd2.chatbot.prose p {{
     color: #266B99;
 }}
-
+/* Chatbot container */
+.gradio-container {{
+    background: #1c1c1c; /* Dark 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 {{
     background: #DB1616;
@@ -89,39 +84,35 @@ span.md.svelte-8tpqd2.chatbot.prose p {{
     content: url('file=Komal-patra/EU_AI_ACT/orcawise image.png');
     margin-right: 8px;
 }}
-
 /* Enable scrolling for the chatbot messages */
 .chatbot .messages {{
     max-height: 500px;  /* Adjust as needed */
     overflow-y: auto;
 }}
-
-/* Transparent overall message container */
-.gradio-chatbot .wrap {{
-    background: transparent; /* Make the container background transparent */
-    border: none; /* Remove any borders if necessary */
-}}
 """
 
 # Gradio interface setup
 with gr.Blocks(css=custom_css) as demo:
+    gr.Markdown("<h1>Ask a question about the EU AI Act</h1>")
     chatbot = gr.Chatbot()
-    msg = gr.Textbox(placeholder="Ask your question...", show_label=False)
+    msg = gr.Textbox(placeholder="Ask your question...", show_label=False)  # Add placeholder text
     submit_button = gr.Button("Submit", elem_classes="submit-btn")
     clear = gr.Button("Clear", elem_classes="clear-btn")
 
+    # Function to handle user input
     def user(user_message, history):
         return "", history + [[user_message, None]]
 
+    # Function to handle bot response
     def bot(history):
-        if len(history) == 1:
+        if len(history) == 1:  # Check if it's the first interaction
             bot_message = "Hi there! How can I help you today?"
-            history[-1][1] = bot_message
+            history[-1][1] = bot_message  # Add welcome message to history
         else:
-            history[-1][1] = ""
-            previous_message = history[-1][0]
-            bot_message = generate_text(previous_message)
-            history[-1][1] = bot_message
+            history[-1][1] = ""  # Clear the last bot message
+            previous_message = history[-1][0]  # Access the previous user message
+            bot_message = generate_text(previous_message)  # Generate response based on previous message
+            history[-1][1] = bot_message  # Update the last bot message
         return history
 
     submit_button.click(user, [msg, chatbot], [msg, chatbot], queue=False).then(