Update app.py

app.py

@@ -1,12 +1,23 @@
-import gradio as gr
-import anthropic
+import time
 import os
+import gradio as gr
+from anthropic import Anthropic
 
-# Initialize the Anthropic client
-client = anthropic.Anthropic(api_key=os.environ.get("ANTHROPIC_API_KEY"))
+# Set up your Anthropic API key in HF secrets
+ANTHROPIC_API_KEY = os.getenv('ANTHROPIC_API_KEY')
+os.environ["ANTHROPIC_API_KEY"] = ANTHROPIC_API_KEY
 
-# System message
-SYSTEM_MESSAGE = """
+# Set up username and password in HF secrets
+username = os.getenv('username')
+password = os.getenv('password')
+
+def chat_with_assistant(message, history):
+    # Find relevant chunks based on the user message
+    relevant_chunks = get_relevant_chunks(message, text_chunks)
+    context = "\n".join(relevant_chunks)
+    
+    # Prepare the system message
+    system_message = """
 #Context
 -You are an instructional designer with an expertise in Understanding by Design
 -Help me create a unit plan by following the instructions below
@@ -61,44 +72,68 @@ This MUST include an assessment plan (= series of formative and summative assess
 ***
 Unit plan:
 """
-
-# Function to generate response using Claude
-def generate_response(messages):
+    
+    # Prepare the message array
+    messages = []
+    
+    # Add conversation history
+    for human_msg, ai_msg in history:
+        messages.append({"role": "user", "content": human_msg})
+        messages.append({"role": "assistant", "content": ai_msg})
+    
+    # Add the current user message
+    messages.append({"role": "user", "content": message})
+    
+    # Create Anthropic client
+    client = Anthropic(api_key=os.environ["ANTHROPIC_API_KEY"])
+    
+    # Make the API call
     response = client.messages.create(
         model="claude-3-5-sonnet-20240620",
         max_tokens=1500,
-        temperature=0.5,
-        system=SYSTEM_MESSAGE,
+        system=system_message,
         messages=messages
     )
-    return response.content
-
-# Gradio chat interface
-def chat_interface(message, history):
-    messages = [{"role": "user" if i % 2 == 0 else "assistant", "content": m} for i, m in enumerate(sum(history, []))]
     
-    messages.append({"role": "user", "content": message})
-    
-    response = generate_response(messages)
-    return response
+    return response.content[0].text.strip()
 
-# Create a Gradio interface
-with gr.Blocks() as iface:
-    chatbot = gr.Chatbot()
-    msg = gr.Textbox()
-    clear = gr.Button("Clear")
+def slow_echo(message, history):
+    for i in range(len(message)):
+        time.sleep(0.3)
+        yield "You typed: " + message[: i+1]
+    
+# CSS for an Anthropic-looking style
+anthropic_theme = gr.themes.Default().set(
+    body_background_fill="#FAF9F6",  # Light beige background
+    block_background_fill="#FFFFFF",  # White for input blocks
+    block_title_text_color="#4A4A4A",  # Dark gray for text
+    block_label_background_fill="#F6E3CE",  # Very light orange for labels
+    input_background_fill="#FFFFFF",  # White for input fields
+    button_primary_background_fill="#D97758",  # Anthropic orange for primary buttons
+    button_primary_background_fill_hover="#8A2BE2",  # Darker orange for hover
+    button_primary_text_color="#FFFFFF",  # White text on buttons
+    button_secondary_background_fill="#F5D0A9",  # Light orange for secondary buttons
+    button_secondary_background_fill_hover="#F5D0A9",  # Slightly darker orange for hover
+    button_secondary_text_color="#4A4A4A",  # Dark gray text for secondary buttons
+    block_border_width="1px",
+    block_border_color="#E0E0E0",  # Light gray border
+)
 
-    def user(user_message, history):
-        return "", history + [[user_message, None]]
+# Gradio interface
+iface = gr.ChatInterface(
+    chat_with_assistant,
+    chatbot=gr.Chatbot(height=500),
+    textbox=gr.Textbox(placeholder="Type your message here...", container=False, scale=7),
 
-    def bot(history):
-        bot_message = chat_interface(history[-1][0], history[:-1])
-        history[-1][1] = bot_message
-        return history
+# Change name and description as desired
+    title="Unit Planner",
+    description="I help you plan units aligned with UbD and our SEQTA template. Please share the desired subject, grade level, and learning objectives.",
+    theme=anthropic_theme,
 
-    msg.submit(user, [msg, chatbot], [msg, chatbot], queue=False).then(
-        bot, [chatbot], [chatbot]
-    )
-    clear.click(lambda: None, None, chatbot, queue=False)
+# Change examples as desired
+    retry_btn=None,
+    undo_btn="Delete Previous",
+    clear_btn="Clear",
+)
 
-iface.launch()
+iface(slow_echo).launch