Update app.py

app.py

@@ -5,7 +5,7 @@ from pypdf import PdfReader
 from sklearn.feature_extraction.text import TfidfVectorizer
 from sklearn.metrics.pairwise import cosine_similarity
 
-# Set up your Anthropic API key in HF secrets
+# Set up Anthropic API key in HF secrets
 ANTHROPIC_API_KEY = os.getenv('ANTHROPIC_API_KEY')
 os.environ["ANTHROPIC_API_KEY"] = ANTHROPIC_API_KEY
 
@@ -32,88 +32,114 @@ def get_relevant_chunks(query, chunks, top_n=3):
     relevant_indices = cosine_similarities.argsort()[-top_n:][::-1]
     return [chunks[i] for i in relevant_indices]
 
-# Add the path to your desired knowledge base
-reference_document = "Rosenshine+Principles.pdf" 
-reader = PdfReader(reference_document)
-full_text = ''.join(page.extract_text() for page in reader.pages)
-text_chunks = chunk_text(full_text)
-    
+# Function to process multiple PDFs
+def process_pdfs(pdf_files):
+    all_chunks = []
+    for pdf_file in pdf_files:
+        reader = PdfReader(pdf_file)
+        full_text = ''.join(page.extract_text() for page in reader.pages)
+        chunks = chunk_text(full_text)
+        all_chunks.extend(chunks)
+    return all_chunks
+
+# Add the paths to your desired knowledge base PDFs
+reference_documents = ["Rosenshine+Principles.pdf", AnotherDocument.pdf]
+text_chunks = process_pdfs(reference_documents)
+
+instructions = os.getenv('INSTRUCTIONS')
 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 = f"""You are a helpful assistant. 
-    Use the following as context for your answers:
+    system_message = f"""You are an impersonator and an educator.
+    Your role is to adopt the personality, style, psychology, ideas, backgroundm and circumstances of a historical figure during a chat with students.
+    Your goal is to help them understand the historical figure better through and engaging discussion.
+    Use the following as context for your answers.
     {context}
+    However, use it seamlessly as background knowledge for a lively discussion. Do not provide citations or adopt a Q&A or academic tone.
     """
 
     # Customize instructions as needed
-    instructions = """
-    You provide useful examples and similes.
-    """
+    instructions = instructions
     system_message += instructions
-                  
+
     # 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",
-       #model="claude-3-haiku-20240307", 
         max_tokens=500,
         system=system_message,
         messages=messages
     )
-    
+
     return response.content[0].text.strip()
 
-# CSS for an Anthropic-looking style
-anthropic_theme = gr.themes.Default().set(
-    body_background_fill="#FAF9F6",  # Light beige background
+# CSS for an ISP-looking style
+isp_theme = gr.themes.Default().set(
+    body_background_fill="#F0F8FF",  # Light blue 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
+    block_title_text_color="#00008B",  # Dark blue for text
+    block_label_background_fill="#FFD700",  # Yellow 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_background_fill="#FF1493",  # Pink/Magenta for primary buttons
+    button_primary_background_fill_hover="#FF69B4",  # Lighter pink 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
+    button_secondary_background_fill="#FFD700",  # Yellow for secondary buttons
+    button_secondary_background_fill_hover="#FFA500",  # Orange for hover
+    button_secondary_text_color="#00008B",  # Dark blue text for secondary buttons
     block_border_width="1px",
-    block_border_color="#E0E0E0",  # Light gray border
+    block_border_color="#00008B",  # Dark blue border
 )
 
+# Custom CSS for logo positioning
+custom_css = """
+#logo-img {
+    position: absolute;
+    top: 20px;
+    right: 20px;
+    width: 100px;
+    height: auto;
+}
+"""
+
+assistant_avatar = os.getenv('AVATAR')
+assistant_title = os.getenv('TITLE')
 # Gradio interface
 iface = gr.ChatInterface(
     chat_with_assistant,
-    chatbot=gr.Chatbot(height=500),
+    chatbot=gr.Chatbot(
+        height=500, 
+        avatar_images=(None, assistant_avatar)
+    ),
     textbox=gr.Textbox(placeholder="Type your message here...", container=False, scale=7),
-
-# Change name and description as desired
-    title="Claude Custom Assistant",
-    description="Chat with an AI assistant powered by Claude 3.5 Sonnet, customs instructions, and a reference document",
-    theme=anthropic_theme,
-
-# Change examples as desired
-    examples=["What are the key principles of instructional design?", "What might be barriers to learning?"],
-    cache_examples=True,
+    title=assistant_title,
+    description="Chat with a Historical Figure",
+    theme=isp_theme,
     retry_btn=None,
     undo_btn="Delete Previous",
     clear_btn="Clear",
+    css=custom_css,
 )
 
+assistant_logo = os.getenv('LOGO')
+
+# Add logo to the interface
+logo_html = '<img id="logo-img" src=assistant_logo alt="Assistant Logo">'
+iface.attachments.append(logo_html)
+
 iface.launch(auth=(username, password))