updated langfuse added authentication

app.py

@@ -3,6 +3,7 @@ import uuid
 import gradio as gr
 from openai import OpenAI
 from langfuse import Langfuse
+from langfuse.decorators import observe
 from dotenv import load_dotenv
 
 # Load environment variables from .env file if it exists
@@ -20,6 +21,7 @@ try:
     langfuse = Langfuse(
         public_key=os.getenv("LANGFUSE_PUBLIC_KEY"),
         secret_key=os.getenv("LANGFUSE_SECRET_KEY"),
+        host=os.getenv("LANGFUSE_HOST")
     )
 except Exception as e:
     print(f"Warning: Langfuse client initialization failed: {e}")
@@ -41,25 +43,38 @@ def create_chat_interface():
         except Exception as e:
             print(f"Warning: Failed to create Langfuse trace: {e}")
 
-    def respond(message, chat_history):
-        if not message:
-            return chat_history, ""
+    @observe(
+        name="chat_completion",
+        capture_input=True,
+        capture_output=True
+    )
+    def get_completion(messages):
+        if not client:
+            raise Exception("OpenAI client not initialized. Please check your API key.")
+        
+        response = client.chat.completions.create(
+            model="gpt-3.5-turbo",
+            messages=messages
+        )
         
-        # Track user message if Langfuse is available
-        span = None
-        if trace:
+        # Add model info to the span
+        if langfuse and trace:
             try:
-                span = trace.span(
-                    name="user_message",
-                    input={"message": message}
-                )
+                current_span = trace.get_current_span()
+                if current_span:
+                    current_span.update(
+                        metadata={"model": "gpt-3.5-turbo", "tokens": response.usage.total_tokens}
+                    )
             except Exception as e:
-                print(f"Warning: Failed to create Langfuse span: {e}")
+                print(f"Warning: Failed to update span metadata: {e}")
         
-        try:
-            if not client:
-                return chat_history + [(message, "Error: OpenAI client not initialized. Please check your API key.")], ""
+        return response.choices[0].message.content
 
+    def respond(message, chat_history):
+        if not message:
+            return chat_history, ""
+        
+        try:
             # Format chat history for OpenAI
             messages = [
                 {"role": "system", "content": """
@@ -128,48 +143,20 @@ Write like the stakes are real. Because they are.
             # Add current message
             messages.append({"role": "user", "content": message})
             
-            # Get response from OpenAI
-            response = client.chat.completions.create(
-                model="gpt-3.5-turbo",
-                messages=messages
-            )
-            
-            assistant_message = response.choices[0].message.content
-            
-            # Track assistant response if Langfuse is available
-            if span:
-                try:
-                    span.end(
-                        output={"response": assistant_message},
-                        metadata={
-                            "model": "gpt-3.5-turbo",
-                            "tokens": response.usage.total_tokens
-                        }
-                    )
-                except Exception as e:
-                    print(f"Warning: Failed to end Langfuse span: {e}")
+            # Get response from OpenAI with Langfuse tracking
+            assistant_message = get_completion(messages)
             
             chat_history.append((message, assistant_message))
             return chat_history, ""
             
         except Exception as e:
-            # End span with error if Langfuse is available
-            if span:
-                try:
-                    span.end(
-                        output={"error": str(e)},
-                        level="ERROR"
-                    )
-                except Exception as span_error:
-                    print(f"Warning: Failed to end Langfuse span with error: {span_error}")
-            
             error_message = f"Error: {str(e)}"
             return chat_history + [(message, error_message)], ""
 
     # Create Gradio interface
     with gr.Blocks() as demo:
-        gr.Markdown("# Rag Bot AI - LLM")
-        gr.Markdown("")
+        gr.Markdown("# Greg Logan AI - Brand Strategy Assistant")
+        gr.Markdown("Get direct, punchy, and provocative brand strategy insights from Greg Logan's perspective.")
         
         chatbot = gr.Chatbot(height=600)
         with gr.Row():
@@ -187,4 +174,12 @@ Write like the stakes are real. Because they are.
 
 # Create and launch the interface
 demo = create_chat_interface()
-demo.launch() 
+
+# Get auth credentials from environment variables
+auth_username = os.getenv("AUTH_USERNAME", "admin")
+auth_password = os.getenv("AUTH_PASSWORD", "admin")
+
+# Launch with authentication
+demo.launch(
+    auth=(auth_username, auth_password)
+)