Renewed

app.py

@@ -1,44 +1,92 @@
 import streamlit as st
-import random
 from openai import OpenAI
+import re
+import os
+
+# Initialize OpenAI client (REPLACE with your actual key from environment variables or a secure config)
+
+# IMPORTANT: Never hardcode API keys in your code, especially for sharing!
 
-# Initialize OpenAI client (replace with your actual key)
 client = OpenAI(
-    base_url="https://integrate.api.nvidia.com/v1",
-    api_key="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"  # Replace with your actual key
+
+base_url="https://integrate.api.nvidia.com/v1", # Assuming this is your correct base URL
+
+api_key="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" # Replace with your actual key
+
 )
 
-st.title("Harry Potter Character Description")
 
-# Input for character name (optional, defaults to Dumbledore)
-character_name = st.text_input("Enter a Harry Potter character name (optional):", "Albus Percival Wulfric Brian Dumbledore")
+st.title("Sorting Hat")
+
+def get_essay_answer():
+    """Gets the essay answer from the user."""
+    st.subheader("Answer the Question:")
+    essay = st.text_area("Describe what you would do if you encountered a lost and distressed first-year student who has lost their pet toad and is terrified of being late for Herbology.", height=200) # Increased height for essay
+    return essay
+
+def get_sorted(essay):
+    """Handles the sorting process using the OpenAI API."""
+    if essay:  # Check if essay is not empty
+        output_area = st.empty()
+
+        messages = [{"role": "user", "content": f"""Pretend that you are the sorting hat in Harry Potter. Sort me into a house in Hogwarts based on my answer. Give me the house ONLY (one word: Gryffindor, Hufflepuff, Ravenclaw, Slytherin) on the first line. On the second line explain your reasoning for sorting me there. Here is my answer:
+        {essay}
+        """}]
+
+        with st.spinner("Sorting..."):
+            try:
+                completion = client.chat.completions.create(
+                    model="meta/llama-3.1-405b-instruct",  # Your model
+                    messages=messages,
+                    temperature=0.7,
+                    max_tokens=200,
+                    stream=True,
+                )
+
+                full_response = ""
+                message_placeholder = st.empty()
 
+                for chunk in completion:
+                    if chunk.choices[0].delta and chunk.choices[0].delta.content:
+                        full_response += chunk.choices[0].delta.content
+                        message_placeholder.markdown(full_response, unsafe_allow_html=False)
+                    elif chunk.choices[0].finish_reason == "stop":
+                        break
+
+                house_match = re.search(r"(Gryffindor|Hufflepuff|Ravenclaw|Slytherin)", full_response, re.IGNORECASE)
+
+                if house_match:
+                    house = house_match.group(0).strip()
+                    try:
+                        parts = full_response.split(house + '\n', 1)
+                        if len(parts) > 1:
+                            explanation = parts[1].strip()
+                        else:
+                            explanation = "No explanation provided."
+                    except Exception as e:
+                        explanation = f"Error extracting explanation: {e}"
+
+                    gif_path = {
+                        "Gryffindor": "https://i.pinimg.com/originals/52/46/eb/5246eba389dfc5722324b484975a012f.gif",
+                        "Hufflepuff": "https://c.tenor.com/ThUF6Bm-GLgAAAAM/always-harry-potter.gif",
+                        "Ravenclaw": "https://media.tenor.com/Sgm7TzrEQzAAAAAC/harry-potter-ravenclaw.gif",
+                        "Slytherin": "https://media.giphy.com/media/A1xhQcogWZAIg/giphy.gif"
+                    }.get(house, "default.gif")
+
+                    output_area.image(gif_path, width=300)
+
+                else:
+                    output_area.error("The Sorting Hat's response was not in the expected format.")
+
+            except Exception as e:
+                output_area.error(f"An error occurred: {e}")
 
-if st.button("Get Description"):
-    if character_name:
-        messages = [{"content": f'{{"fullName":"{character_name}"}}', "role": "tool", "tool_call_id": "chatcmpl-tool-c64c70cc341547abbf96bb98573cc3b9", "name": "describe_harry_potter_character"}]
     else:
-        messages = [{"content": '{"fullName":"Albus Percival Wulfric Brian Dumbledore"}', "role": "tool", "tool_call_id": "chatcmpl-tool-c64c70cc341547abbf96bb98573cc3b9", "name": "describe_harry_potter_character"}]
-
-
-    with st.spinner("Generating description..."):  # Show a spinner while generating
-        completion = client.chat.completions.create(
-            model="meta/llama-3.1-405b-instruct",
-            messages=messages,
-            temperature=0.2,
-            top_p=0.7,
-            max_tokens=1024,
-            stream=True,
-            tools=[{"type": "function", "function": {"name": "describe_harry_potter_character", "description": "Returns information and images of Harry Potter characters.", "parameters": {"type": "object", "properties": {"name": {"type": "string", "enum": ["Harry James Potter", "Hermione Jean Granger", "Ron Weasley", "Fred Weasley", "George Weasley", "Bill Weasley", "Percy Weasley", "Charlie Weasley", "Ginny Weasley", "Molly Weasley", "Arthur Weasley", "Neville Longbottom", "Luna Lovegood", "Draco Malfoy", "Albus Percival Wulfric Brian Dumbledore", "Minerva McGonagall", "Remus Lupin", "Rubeus Hagrid", "Sirius Black", "Severus Snape", "Bellatrix Lestrange", "Lord Voldemort", "Cedric Diggory", "Nymphadora Tonks", "James Potter"], "description": "Name of the Harry Potter character"}}, "required": ["name"]}}}],
-            tool_choice="auto"
-        )
-
-        full_response = ""  # Accumulate the streamed response
-        for chunk in completion:
-            if chunk.choices[0].delta.content is not None:
-                full_response += chunk.choices[0].delta.content
-                st.write(chunk.choices[0].delta.content, unsafe_allow_html=False) # Display each chunk as it arrives
-
-
-        # You can now use the full_response if needed after the loop
-        # st.write("Full Response:", full_response) #For debugging
+        st.warning("Please provide an answer.")
+
+
+# Main app flow
+essay = get_essay_answer()  # Get the essay answer
+
+if st.button("Get Sorted!"):
+    get_sorted(essay)  # Pass the essay to get_sorted()