Update app.py

app.py

@@ -1,4 +1,4 @@
-# app.py
+# app.py (with 1024-character limit)
 import gradio as gr
 import torch
 from transformers import AutoTokenizer, AutoModelForCausalLM
@@ -24,11 +24,11 @@ if os.path.exists(CACHE_FILE):
         cache = json.load(f)
 
 def code_assistant(prompt, language):
-    # Input validation
+    # Input validation with a 1024-character limit
     if not prompt.strip():
-        return "Error: The input prompt cannot be empty. Please provide a coding question or code snippet."
-    if len(prompt) > 256:
-        return "Error: The input prompt is too long. Please limit it to 256 characters."
+        return "⚠️ Error: The input prompt cannot be empty. Please provide a coding question or code snippet."
+    if len(prompt) > 1024:
+        return "⚠️ Error: The input prompt is too long. Please limit it to 1024 characters."
 
     # Check if the prompt is in cache
     cache_key = (prompt, language)
@@ -45,7 +45,7 @@ def code_assistant(prompt, language):
     # Generate response with adjusted parameters for faster CPU response
     outputs = model.generate(
         inputs.input_ids,
-        max_length=128,        # Shortened max length for quicker response
+        max_length=256,        # Shortened max length for quicker response
         temperature=0.1,       # Lower temperature for focused output
         top_p=0.8,             # Slightly reduced top_p for quicker sampling
         do_sample=True
@@ -65,16 +65,79 @@ def code_assistant(prompt, language):
 
     return generated_text
 
+# Custom CSS styling for animations and colors
+css = """
+/* Center-align all text in the input and output boxes */
+input, textarea, .output_text {
+    text-align: center;
+}
+
+/* Style the main title */
+h1 {
+    color: #1e90ff;
+    font-family: 'Arial', sans-serif;
+    text-align: center;
+    font-weight: bold;
+}
+
+/* Style the description */
+.description {
+    color: #555;
+    font-family: 'Arial', sans-serif;
+    text-align: center;
+    margin-bottom: 20px;
+}
+
+/* Output box animation */
+.output_text {
+    color: #1e90ff;
+    animation: fadeIn 2s ease-in-out;
+}
+
+/* Add fade-in animation */
+@keyframes fadeIn {
+    0% { opacity: 0; }
+    100% { opacity: 1; }
+}
+
+/* Hover effect for the submit button */
+button {
+    background-color: #1e90ff;
+    color: white;
+    font-weight: bold;
+    border: none;
+    padding: 10px 20px;
+    border-radius: 5px;
+    transition: background-color 0.3s ease;
+}
+
+button:hover {
+    background-color: #104e8b;
+    cursor: pointer;
+}
+"""
+
+# Enhanced title and description with HTML styling
+title_html = """
+<h1>💻 CodeBand: AI Code Assistant</h1>
+"""
+
+description_html = """
+<p class="description">An AI-powered assistant for coding queries, debugging, and code generation. 
+Choose a programming language for more tailored responses. Limited to 1024 characters.</p>
+"""
+
 # Set up Gradio interface with a dropdown for programming language selection
 iface = gr.Interface(
     fn=code_assistant,
     inputs=[
-        gr.Textbox(lines=5, placeholder="Ask a coding question or paste your code here..."),
+        gr.Textbox(lines=5, placeholder="Ask a coding question or paste your code here...", max_chars=1024),
         gr.Dropdown(choices=["Python", "JavaScript", "Java", "C++", "HTML", "CSS", "SQL", "Other"], label="Programming Language")
     ],
     outputs="text",
-    title="CodeBand",
-    description="An AI code assistant to help you with coding queries, debugging, and code generation. Specify the programming language for more accurate responses."
+    title=title_html,
+    description=description_html,
+    css=css  # Add custom CSS
 )
 
 # Launch the Gradio app