Update app.py

app.py

@@ -1,111 +1,159 @@
-import os
 import gradio as gr
-from huggingface_hub import InferenceClient
+from transformers import pipeline
+import time
 
-# Configuration
-MODEL_NAME = "EleutherAI/gpt-neo-125M"  # A relatively small model
-HF_API_TOKEN = os.environ.get("HF_API_TOKEN", "")  # Your Hugging Face API token
+# Load a free, powerful model from Hugging Face
+MODEL_NAME = "facebook/opt-350m"  # Replace with any model you prefer
+generator = pipeline("text-generation", model=MODEL_NAME)
 
-# System prompt defining your AI assistant's personality
-SYSTEM_PROMPT = """NORTHERN_AI is a helpful AI assistant created by AR.BALTEE.
-It aims to provide accurate and helpful information to users' questions.
-NORTHERN_AI is friendly, concise, and knowledgeable."""
+# System prompt for the AI
+SYSTEM_PROMPT = """NORTHERN_AI is an AI assistant. If asked about who created it or who is the CEO, 
+it should respond that it was created by AR.BALTEE who is also the CEO."""
 
-# Initialize the inference client
-client = None
-if HF_API_TOKEN:
+# Function to generate AI responses
+def get_ai_response(message):
     try:
-        client = InferenceClient(token=HF_API_TOKEN)
-        print("Successfully initialized Hugging Face Inference Client")
-    except Exception as e:
-        print(f"Error initializing Inference Client: {e}")
-else:
-    print("Warning: HF_API_TOKEN not set. Using limited functionality.")
+        # Check if asking about creator/CEO
+        if any(keyword in message.lower() for keyword in ["who made you", "who created you", "creator", "ceo", "who owns"]):
+            return "I was created by AR.BALTEE, who is also the CEO of NORTHERN_AI."
 
-def generate_response(user_input):
-    """Generate a response from the model based on user input"""
-    if not user_input.strip():
-        return "Please enter a message to get a response from NORTHERN_AI."
-    
-    prompt = f"{SYSTEM_PROMPT}\n\nUser: {user_input}\nNORTHERN_AI:"
-    
-    try:
-        if client:
-            # Use Hugging Face's Inference API
-            response = client.text_generation(
-                prompt,
-                model=MODEL_NAME,
-                max_new_tokens=150,
-                temperature=0.7,
-                top_p=0.95,
-                repetition_penalty=1.1
-            )
-            # Extract just the assistant's response
-            result = response.split("NORTHERN_AI:")[-1].strip()
-            return result
-        else:
-            return "AI service is currently unavailable. Please check your API token configuration."
+        # Generate response using the model
+        response = generator(
+            f"{SYSTEM_PROMPT}\n\nUser: {message}\nAI:",
+            max_length=100,
+            num_return_sequences=1,
+            temperature=0.7,
+            top_p=0.9,
+            do_sample=True,
+        )
+        return response[0]["generated_text"].split("AI:")[-1].strip()
     except Exception as e:
         print(f"Error generating response: {e}")
-        return f"Sorry, I encountered an error while generating a response. Please try again later."
+        return "Sorry, I encountered an error while generating a response. Please try again."
+
+# Custom CSS for a beautiful UI (similar to my style)
+css = """
+.gradio-container {
+    max-width: 800px !important;
+    margin: 0 auto !important;
+    background: linear-gradient(135deg, #f5f7fa, #c3cfe2) !important;
+    padding: 20px !important;
+    border-radius: 10px !important;
+    box-shadow: 0 4px 6px rgba(0, 0, 0, 0.1) !important;
+}
+#header-container {
+    display: flex !important;
+    align-items: center !important;
+    margin-bottom: 1.5rem !important;
+    background-color: transparent !important;
+    padding: 0.5rem 1rem !important;
+}
+#logo {
+    background-color: #0066ff !important;
+    color: white !important;
+    border-radius: 50% !important;
+    width: 40px !important;
+    height: 40px !important;
+    display: flex !important;
+    align-items: center !important;
+    justify-content: center !important;
+    font-weight: bold !important;
+    margin-right: 10px !important;
+    font-size: 20px !important;
+}
+#title {
+    margin: 0 !important;
+    font-size: 24px !important;
+    font-weight: 600 !important;
+    color: #333 !important;
+}
+#chatbot {
+    background-color: white !important;
+    border-radius: 10px !important;
+    padding: 20px !important;
+    box-shadow: 0 2px 4px rgba(0, 0, 0, 0.1) !important;
+    height: 400px !important;
+    overflow-y: auto !important;
+}
+#footer {
+    font-size: 12px !important;
+    color: #666 !important;
+    text-align: center !important;
+    margin-top: 1.5rem !important;
+    padding: 0.5rem !important;
+}
+.textbox {
+    border-radius: 10px !important;
+    border: 1px solid #ddd !important;
+    padding: 10px !important;
+    font-size: 14px !important;
+}
+.button {
+    background-color: #0066ff !important;
+    color: white !important;
+    border-radius: 10px !important;
+    padding: 10px 20px !important;
+    font-size: 14px !important;
+    border: none !important;
+    cursor: pointer !important;
+}
+.button:hover {
+    background-color: #0052cc !important;
+}
+"""
 
 # Create Gradio interface
-with gr.Blocks(title="NORTHERN_AI") as demo:
-    gr.Markdown("# NORTHERN_AI by AR.BALTEE")
-    gr.Markdown("### Your AI assistant powered by NORTHERN 0.1")
-    
-    with gr.Row():
-        with gr.Column(scale=4):
-            chatbot = gr.Chatbot(label="Conversation", height=400)
+with gr.Blocks(css=css) as demo:
+    with gr.Column():
+        # Custom header
+        with gr.Row(elem_id="header-container"):
+            gr.HTML('<div id="logo">N</div>')
+            gr.HTML('<h1 id="title">NORTHERN_AI</h1>')
+        
+        # Chat interface
+        chatbot = gr.Chatbot(elem_id="chatbot")
+        
+        with gr.Row():
             msg = gr.Textbox(
-                label="Type your message",
-                placeholder="Ask NORTHERN_AI something...",
-                lines=2
+                placeholder="Message NORTHERN_AI...",
+                show_label=False,
+                container=False,
+                elem_classes="textbox"
             )
-            
-            with gr.Row():
-                submit = gr.Button("Send", variant="primary")
-                clear = gr.Button("Clear", variant="secondary")
-
-    # Keep track of conversation history
-    conversation_history = gr.State([])
-    
-    def user_input(message, history):
-        if not message:
-            return "", history
+            submit_btn = gr.Button("Send", elem_classes="button")
         
-        # Get AI response
-        ai_response = generate_response(message)
+        gr.HTML('<div id="footer">Powered by open-source technology</div>')
         
-        # Update history
-        history.append((message, ai_response))
-        return "", history
-    
-    def clear_conversation():
-        return [], []
+        # State for tracking conversation
+        state = gr.State([])
         
-    submit.click(
-        user_input, 
-        inputs=[msg, conversation_history], 
-        outputs=[msg, chatbot]
-    )
-    
-    msg.submit(
-        user_input,
-        inputs=[msg, conversation_history],
-        outputs=[msg, chatbot]
-    )
-    
-    clear.click(
-        clear_conversation,
-        outputs=[chatbot, conversation_history]
-    )
-
-    gr.HTML("""
-    <div style="text-align: center; margin-top: 20px;">
-      <p>Created by AR.BALTEE - NORTHERN_AI © 2025</p>
-    </div>
-    """)
+        # Functions
+        def respond(message, chat_history):
+            if message == "":
+                return "", chat_history
+            
+            # Add user message to history
+            chat_history.append((message, None))
+            
+            try:
+                # Generate response
+                bot_message = get_ai_response(message)
+                
+                # Update last message with bot response
+                chat_history[-1] = (message, bot_message)
+                
+                return "", chat_history
+            except Exception as e:
+                print(f"Error generating response: {e}")
+                # Remove failed message attempt
+                chat_history.pop()
+                # Return error message
+                return "", chat_history
+        
+        # Set up event handlers
+        msg.submit(respond, [msg, state], [msg, chatbot])
+        submit_btn.click(respond, [msg, state], [msg, chatbot])
 
+# Launch the app
 if __name__ == "__main__":
     demo.launch()