Update app.py

app.py

@@ -49,125 +49,63 @@ def generate_response(user_input):
         print(f"Error generating response: {e}")
         return f"Sorry, I encountered an error while generating a response. Please try again later."
 
-# For demonstration purposes, we'll use a simpler model approach
-# that doesn't require an API key but still provides responses
-def get_ai_response(message):
-    """Generate a response using a local model"""
-    # 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."
-    
-    # Simple response generation (replace this with actual model inference)
-    responses = [
-        "I understand your question about {}. Based on my knowledge, I would say that...",
-        "Thanks for asking about {}. Here's what I know on this topic...",
-        "Regarding {}, I can provide some information...",
-        "I've analyzed your question about {} and here's my perspective...",
-        "When it comes to {}, there are several important points to consider..."
-    ]
-    
-    # Extract key topic from message
-    topic = message.split()[0] if len(message.split()) > 0 else "that"
-    
-    # Add a slight delay to simulate processing
-    time.sleep(0.5)
-    
-    return random.choice(responses).format(topic)
-
-# Custom CSS for styling
-css = """
-.gradio-container {
-    max-width: 800px !important;
-    margin: 0 auto !important;
-}
-#header-container {
-    display: flex !important;
-    align-items: center !important;
-    margin-bottom: 1rem !important;
-    background-color: black !important;
-    padding: 0.5rem 1rem !important;
-    border-bottom: 1px solid #eee !important;
-}
-#logo {
-    background-color: #0066ff !important;
-    color: white !important;
-    border-radius: 50% !important;
-    width: 32px !important;
-    height: 32px !important;
-    display: flex !important;
-    align-items: center !important;
-    justify-content: center !important;
-    font-weight: bold !important;
-    margin-right: 10px !important;
-}
-#title {
-    margin: 0 !important;
-    font-size: 18px !important;
-    font-weight: 600 !important;
-}
-#footer {
-    font-size: 12px !important;
-    color: #666 !important;
-    text-align: center !important;
-    margin-top: 1rem !important;
-    padding: 0.5rem !important;
-}
-"""
-
 # Create Gradio interface
-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(
-            height=400,
-            show_label=False,
-        )
-        
-        with gr.Row():
+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)
             msg = gr.Textbox(
-                placeholder="Message NORTHERN_AI...",
-                show_label=False,
-                container=False
+                label="Type your message",
+                placeholder="Ask NORTHERN_AI something...",
+                lines=2
             )
-            submit_btn = gr.Button("Send")
-        
-        gr.HTML('<div id="footer">Powered by open-source technology</div>')
+            
+            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
         
-        # State for tracking conversation
-        state = gr.State([])
+        # Get AI response
+        ai_response = generate_response(message)
         
-        # 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
+        # Update history
+        history.append((message, ai_response))
+        return "", history
+    
+    def clear_conversation():
+        return [], []
         
-        # Set up event handlers
-        msg.submit(respond, [msg, state], [msg, chatbot])
-        submit_btn.click(respond, [msg, state], [msg, chatbot])
+    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>
+    """)
 
-# Launch the app
 if __name__ == "__main__":
     demo.launch()