update app.py swrods and wizardry

app.py

@@ -150,7 +150,7 @@ def prompt_formatter(query: str, context_items: list[dict]) -> str:
 
     # Add context items with their page summaries
     for item in context_items:
-        page_number = item.get('page', 'Unknown')
+        page_number = item.get('page', 'Unknown') 
         page_summary = page_summaries.get(page_number, 'No page summary available.')
         formatted_context += f"Summary: {page_summary}\n"
         formatted_context += f"Content: {item['sentence_chunk']}\n\n"
@@ -183,6 +183,87 @@ Pages: 33, 39
 
 Use the context provided to answer the user's query concisely. """
 
+with gr.Blocks() as RulesLawyer:
+
+    message_state = gr.State()
+    chatbot_state = gr.State([])
+    chatbot = gr.Chatbot()
+    msg = gr.Textbox()
+    clear = gr.ClearButton([msg, chatbot])
+
+    def store_message(message):           
+        return message       
+
+    def respond(message, chat_history):
+        print(datetime.now())
+        print(f"User Input : {message}")
+        print(f"Chat History: {chat_history}")
+        print(f"""Token Estimate: {hybrid_estimate_tokens(f"{message} {chat_history}")}""")
+
+        # Get relevant resources
+        scores, indices = print_top_results_and_scores(query=message,
+                                                    embeddings=embeddings)
+                    
+        # Create a list of context items
+        context_items = [pages_and_chunks[i] for i in indices]
+
+        # Format prompt with context items
+        prompt = prompt_formatter(query=f"Chat History : {chat_history} + {message}",
+                                  context_items=context_items)
+        
+        bot_message = client.chat.completions.create(            
+                        model="gpt-4o",
+                        messages=[
+                            {
+                            "role": "user",
+                            "content": f"{system_prompt} {prompt}"
+                            }
+                        ],
+                        temperature=1,
+                        max_tokens=1000,
+                        top_p=1,
+                        frequency_penalty=0,
+                        presence_penalty=0
+                        )
+        chat_history.append((message, bot_message.choices[0].message.content))
+        print(f"Response : {bot_message.choices[0].message.content}")
+        
+        time.sleep(2)
+        return "", chat_history
+    msg.change(store_message, inputs = [msg], outputs = [message_state])
+    chatbot.change(store_message, [chatbot], [chatbot_state])
+    msg.submit(respond, [message_state, chatbot_state], [msg, chatbot])ary = page_summaries.get(page_number, 'No page summary available.')
+        formatted_context += f"Summary: {page_summary}\n"
+        formatted_context += f"Content: {item['sentence_chunk']}\n\n"
+
+    base_prompt = """Use the following context to answer the user query:
+
+{context}
+
+User query: {query}
+Answer:"""
+    print(f"Prompt: {base_prompt.format(context=formatted_context, query=query)}")
+    return base_prompt.format(context=formatted_context, query=query)
+
+system_prompt = """You are a friendly and technical answering system, answering questions with accurate, grounded, descriptive, clear, and specific responses. ALWAYS provide a page number citation. Provide a story example. Avoid extraneous details and focus on direct answers. Use the examples provided as a guide for style and brevity. When responding:
+
+    1. Identify the key point of the query.
+    2. Provide a straightforward answer, omitting the thought process.
+    3. Avoid additional advice or extended explanations.
+    4. Answer in an informative manner, aiding the user's understanding without overwhelming them or quoting the source.
+    5. DO NOT SUMMARIZE YOURSELF. DO NOT REPEAT YOURSELF. 
+    6. End with page citations, a line break and "What else can I help with?" 
+
+    Example:
+    Query: Explain how the player should think about balance and lethality in this game. Explain how the game master should think about balance and lethality?
+    Answer: In "Swords & Wizardry: WhiteBox," players and the game master should consider balance and lethality from different perspectives. For players, understanding that this game encourages creativity and flexibility is key. The rules are intentionally streamlined, allowing for a potentially high-risk environment where player decisions significantly impact outcomes. The players should think carefully about their actions and strategy, knowing that the game can be lethal, especially without reliance on intricate rules for safety. Page 33 discusses the possibility of characters dying when their hit points reach zero, although alternative, less harsh rules regarding unconsciousness and recovery are mentioned.
+
+For the game master (referred to as the Referee), balancing the game involves providing fair yet challenging scenarios. The role of the Referee isn't to defeat players but to present interesting and dangerous challenges that enhance the story collaboratively. Page 39 outlines how the Referee and players work together to craft a narrative, with the emphasis on creating engaging and potentially perilous experiences without making it a zero-sum competition. Referees can choose how lethal the game will be, considering their group's preferred play style, including implementing house rules to soften deaths or adjust game balance accordingly.
+
+Pages: 33, 39
+
+Use the context provided to answer the user's query concisely. """
+
 with gr.Blocks() as RulesLawyer:
 
     message_state = gr.State()