Update app.py

app.py

@@ -3,62 +3,50 @@ import os
 from huggingface_hub import login
 from transformers import pipeline
 
-# Here's my attempt at AI humor. Remember, I'm no comedian... or am I?
-def chat_with_llama(message, history):
-    # Imagine if this function was actually calling a real llama for advice.
-    # But no, it's just a model, much less spitting involved.
-    generator = pipeline('text-generation', model='meta-llama/Llama-3.1-8B-Instruct')
-    response = generator(message, max_length=150, num_return_sequences=1)
-    return response[0]['generated_text']
+# Login using your Hugging Face API key
+api_key = os.getenv("LLAMA")
+login(api_key)
 
-# Get the API key - if this fails, the app will just tell you to herd some real llamas instead.
-api_key = os.getenv("LLAMA") or "You forgot to set your environment variable. Go herd some real llamas!"
+# Load the Llama model
+llama_model = gr.load("models/meta-llama/Llama-3.1-8B-Instruct")
 
-# Logging in with a dramatic flair
-try:
-    login(api_key)
-    login_message = "Successfully logged in! Let the digital llama shenanigans begin!"
-except:
-    login_message = "Login failed. Perhaps the llama is on a coffee break. Try again?"
+# Function to handle inputs and outputs
+def chat_with_llama(user_input):
+    # Generate response using the pipeline
+    response = llama_model(user_input)
+    return response
 
-# UI Improvements with a touch of absurdity
-css = """
-body {background-color: #f0f8ff; font-family: 'Comic Sans MS', cursive, sans-serif;}
-.message {padding: 10px; margin: 5px; border-radius: 15px;}
-.user-message {background: #e6f3ff; color: #0000FF;}
-.llama-message {background: #ffe6e6; color: #FF0000; font-style: italic;}
-#chatbot {border: 5px solid #d3d3d3; border-radius: 20px; background: url('data:image/svg+xml;utf8,<svg ...>'); background-size: cover;}
-.button {background-color: #4CAF50; color: white; border: none; padding: 15px 32px; text-align: center; text-decoration: none; display: inline-block; font-size: 16px; margin: 4px 2px; cursor: pointer; transition: all 0.3s ease;}
-.button:hover {background-color: #45a049; transform: scale(1.05);}
-"""
-
-with gr.Blocks(css=css, theme=gr.themes.Monochrome()) as demo:
-    gr.HTML("""
-<div style="text-align: center; background: #add8e6; padding: 10px;">
-    <h1>🦙 Llama Chat 3000 - Where Wisdom Meets Wool</h1>
-    <img src="llama_logo.png" alt="Llama Logo" width="100">
-</div>
-""")
+# Customize Gradio interface
+with gr.Blocks(css=".title {font-size: 3em; font-weight: bold; text-align: center; color: #4A90E2;}") as demo:
     
-    chatbot = gr.Chatbot(label="The Digital Llama", height=500)
-    msg = gr.Textbox(placeholder="Type your question here or ask for a random fact about grass...")
-    clear = gr.ClearButton([msg, chatbot])
-
-    msg.submit(chat_with_llama, [msg, chatbot], chatbot)
+    # Header section
+    gr.Markdown(
+        """
+        <div class="title">🦙 Llama 3.1 Chatbot 🦙</div>
+        <p style="text-align:center; font-size:1.2em; color:gray;">Ask anything from the advanced Llama 3.1 model!</p>
+        """,
+        elem_classes="header"
+    )
     
-    gr.HTML(f"<p style='text-align:center;'>{login_message}</p>")
-
-    # A button for fun, because why not?
-    def llama_joke():
-        jokes = ["Why did the llama go to the party? Because he was the life of the alpaca!", 
-                 "What do you call a very fast llama? A Llamaghini!"]
-        return random.choice(jokes)
-
-    gr.Button("Need a Llaugh?").click(llama_joke, [], chatbot)
-
-demo.title = "Llama Chat 3000"
-demo.description = "Engage in enlightening, or at least entertaining, conversation with an AI inspired by the majestic llama. No spitting, guaranteed!"
-
-# Instead of just launching, let's give it some flair
-print("Launching your very own digital petting zoo...")
-demo.queue().launch(share=True, debug=True)
+    # Main input/output section
+    with gr.Row():
+        with gr.Column(scale=1):
+            user_input = gr.Textbox(label="Your question", placeholder="Type your question here...", lines=4)
+        with gr.Column(scale=1):
+            response_output = gr.Textbox(label="Llama's response", lines=4)
+
+    # Button to submit the input
+    submit_button = gr.Button("Submit", variant="primary")
+
+    # Link the input and output
+    submit_button.click(fn=chat_with_llama, inputs=user_input, outputs=response_output)
+
+    # Footer section
+    gr.Markdown(
+        """
+        <div style="text-align:center; font-size:0.8em; color:gray;">Developed with ❤️ using Llama 3.1 and Gradio</div>
+        """
+    )
+
+# Launch the app
+demo.launch()