Update app.py

app.py

@@ -1,11 +1,5 @@
-!pip install transformers torch gradio
 
-# prompt: code to login to hugging face
 
-import os
-from huggingface_hub import notebook_login
-
-notebook_login()
 
 import torch
 from transformers import AutoTokenizer, AutoModelForCausalLM
@@ -108,56 +102,77 @@ def generate_custom_caption(day, task):
     caption = caption[len(prompt):].strip()
     return caption
 
-# Gradio Interface function to handle marketing schedule and captions, with conversation
 def gradio_interface(user_prompt):
     conversation = []
 
     # Generate the marketing schedule
-    conversation.append("Hassan - Marketing Dept.: I'm processing your request. Please wait a moment...")
-    yield "\n".join(conversation)  # Update the interface step by step
+    conversation.append('<span style="color: #3498db; font-weight: bold;">Hassan - Marketing Dept.:</span> I\'m processing your request. Please wait a moment...')
+    yield "<br>".join(conversation)  # Update the interface with each message on a new line
 
     schedule = generate_marketing_schedule(user_prompt)
 
     if isinstance(schedule, dict):
-        conversation.append("Hassan - Marketing Dept.: I have created the marketing schedule based on the user's request.")
-        yield "\n".join(conversation)  # Update the interface with schedule processing
+        conversation.append('<span style="color: #3498db; font-weight: bold;">Hassan - Marketing Dept.:</span> I have created the marketing schedule based on your request.')
+        yield "<br>".join(conversation)  # Update the interface with schedule processing
 
-        # Add some pause to simulate conversation
         time.sleep(2)
 
-        conversation.append("Hassan Marketing Dept.: Let's now consult the Caption Generator to create captions for each day.")
-        yield "\n".join(conversation)  # Update the interface with new dialogue
+        conversation.append('<span style="color: #3498db; font-weight: bold;">Hassan - Marketing Dept.:</span> Let\'s now consult the Caption Generator to create captions for each day.')
+        yield "<br>".join(conversation)  # Update the interface with new dialogue
 
-        # For each day in the schedule, generate a caption
         captions = {}
         for day, task in schedule.items():
-            conversation.append(f"Hassan Marketing Dept.: For {day}, the task is '{task}'. Caption Generator, can you create a caption?")
-            yield "\n".join(conversation)  # Update the interface with the conversation
+            conversation.append(f'<span style="color: #3498db; font-weight: bold;">Hassan - Marketing Dept.:</span> For <strong>{day}</strong>, the task is <em>\'{task}\'</em>. Caption Generator, can you create a caption?')
+            yield "<br>".join(conversation)  # Update the interface with the conversation
 
-            time.sleep(2)  # Simulate the time taken by the caption generator
+            time.sleep(2)
 
             caption = generate_custom_caption(day, task)
             captions[day] = caption
 
-            conversation.append(f"Ahmed - Media Dept.: Here's the caption for {day}: '{caption}'")
-            yield "\n".join(conversation)  # Update the interface with the caption for the day
+            conversation.append(f'<span style="color: #e74c3c; font-weight: bold;">Ahmed - Media Dept.:</span> Here\'s the caption for <strong>{day}</strong>: <em>\'{caption}\'</em>')
+            yield "<br>".join(conversation)  # Update the interface with the caption for the day
 
-        # Final output
-        conversation.append("Hassan - Marketing Dept.: The final marketing schedule with captions is ready.")
-        yield "\n".join(conversation)  # Final update
+        conversation.append('<span style="color: #3498db; font-weight: bold;">Hassan - Marketing Dept.:</span> The final marketing schedule with captions is ready.')
+        yield "<br>".join(conversation)  # Final update
 
     else:
         yield schedule  # If there was an error, just return the error message
 
-# Launch Gradio interface
+
+# Launch Gradio interface with improved design
 interface = gr.Interface(
     fn=gradio_interface,
-    inputs=gr.Textbox(label="Enter Your Marketing Prompt", placeholder="e.g., Campaign for a new hoodie release."),
-    outputs=gr.Textbox(label="Generated Marketing Schedule & Captions"),
-    title="AI Marketing Schedule and Captions Generator",
-    description="Type a description of your campaign to get a tailored weekly marketing schedule and social media captions.",
-    allow_flagging="never",  # Prevent flagging issues
-    theme="huggingface"  # Use Hugging Face theme
+    inputs=gr.Textbox(label="Enter Your Marketing Prompt", placeholder="e.g., Campaign for a new hoodie release.", lines=2),
+    outputs=gr.Markdown(label="Generated Marketing Schedule & Captions"),  # Changed this to Markdown
+    title="✨ laibor Marketing Demo ✨",
+    description="🧑‍💻 Enter a campaign description to generate a weekly marketing schedule with captions tailored to your brand.",
+    theme="compact",  # Sleek compact theme
+    css="""
+        .gradio-container {
+            background-color: #000000;
+            color: white;
+            font-family: 'Arial', sans-serif;
+            border-radius: 10px;
+            border: 2px solid #dee2e6;
+        }
+        .gr-button {
+            background-color: #4CAF50;
+            color: white;
+            font-size: 16px;
+            font-weight: bold;
+        }
+        .gr-textbox, .gr-markdown {
+            background-color: #333333;
+            color: white;
+            border: 1px solid #555555;
+            border-radius: 8px;
+        }
+        .gr-textbox input, .gr-markdown textarea {
+            background-color: #333333;
+            color: white;
+        }
+    """
 )
 
 interface.launch()