Update app.py

app.py

@@ -1,53 +1,45 @@
 import streamlit as st
 from teapotai import TeapotAI, TeapotAISettings
 import hashlib
+import os
+import requests
 
-default_documents = default_documents = [
-    "The Eiffel Tower is located in Paris, France. It was built in 1889 and stands 330 meters tall. The tower was originally constructed for the 1889 Exposition Universelle (World’s Fair) and was the tallest man-made structure in the world at the time. Designed by Gustave Eiffel, the tower is made of wrought iron and has become a symbol of French engineering and artistry. Visitors from all over the world come to admire its intricate design and take in the panoramic views of Paris from its observation decks.",
-    
-    "The Great Wall of China is a historic fortification that stretches over 13,000 miles. Originally built to protect Chinese states from invading forces, it was constructed over several centuries, with the earliest sections dating back to the 7th century BC. The wall is an extraordinary feat of ancient engineering, featuring watchtowers, barracks, and defensive gates. Today, it stands as one of the most famous landmarks in the world, attracting millions of visitors each year.",
-    
-    "The Amazon Rainforest is the largest tropical rainforest in the world, covering over 5.5 million square kilometers. Spanning across nine countries in South America, the rainforest is home to an incredible array of plant and animal species, many of which are found nowhere else on Earth. The Amazon plays a vital role in regulating the global climate by producing oxygen and absorbing carbon dioxide. However, it is facing significant threats from deforestation and climate change.",
-    
-    "The Grand Canyon is a natural landmark located in Arizona, USA, carved by the Colorado River. It stretches 277 miles long, up to 18 miles wide, and reaches depths of over a mile. The Grand Canyon’s vibrant rock layers reveal millions of years of geological history, making it a fascinating destination for geologists and tourists alike. The canyon's breathtaking views and unique landscape make it one of the most visited natural attractions in the world.",
-    
-    "Mount Everest is the tallest mountain on Earth, located in the Himalayas along the border between Nepal and China. Standing at 8,848 meters above sea level, its peak is known as ‘the Roof of the World.’ Climbing Mount Everest is considered one of the most challenging feats in the world due to its extreme altitude, harsh weather, and difficult terrain. It continues to draw mountaineers and adventurers from across the globe, despite the many risks associated with summiting the mountain.",
-    
-    "The Colosseum in Rome, Italy, is an ancient amphitheater known for its gladiator battles. Completed in 80 AD, the Colosseum was used for a variety of public spectacles, including animal hunts, mock naval battles, and executions. With a capacity to seat up to 50,000 spectators, it was the largest amphitheater in the Roman Empire. Today, it remains a symbol of Roman engineering and entertainment and attracts millions of visitors annually.",
-    
-    "The Sahara Desert is the largest hot desert in the world, located in North Africa. Spanning over 9 million square kilometers, it is known for its vast sand dunes, extreme temperatures, and minimal rainfall. Despite the harsh conditions, the Sahara is home to a variety of life, including desert plants, animals, and nomadic human populations. The desert’s landscape is a stunning mix of rocky plateaus, vast sand dunes, and occasional oases.",
-    
-    "The Nile River is the longest river in the world, flowing through northeastern Africa. It stretches over 6,600 kilometers and has been a vital resource for the civilizations of Egypt and Sudan for thousands of years. The river provides fertile land for agriculture in an otherwise arid region and has played a significant role in the development of these ancient cultures. Today, it remains an essential lifeline for the millions of people who live along its banks.",
-    
-    "The Empire State Building is an iconic skyscraper in New York City that was completed in 1931 and stands at 1454 feet tall. Designed in the Art Deco style by architect William F. Lamb, it was the tallest building in the world until 1970. The Empire State Building features observation decks that offer stunning views of the city and remains an enduring symbol of New York’s skyline. It continues to attract millions of visitors from around the world each year.",
-    
-    "The Pyramids of Giza are a group of ancient Egyptian structures located on the outskirts of Cairo. The Great Pyramid, built for Pharaoh Khufu, is the largest and one of the Seven Wonders of the Ancient World. These massive stone structures were constructed as tombs for the Pharaohs and their queens. The pyramids, along with the nearby Sphinx, are one of Egypt’s most famous tourist attractions, drawing visitors from across the globe.",
-    
-    "The Taj Mahal is a stunning white marble mausoleum located in Agra, India. Built by Emperor Shah Jahan in memory of his wife Mumtaz Mahal, it is considered one of the finest examples of Mughal architecture. The Taj Mahal is known for its intricate designs, including detailed carvings and inlaid gemstones. It is a UNESCO World Heritage site and attracts millions of visitors annually for its beauty and historical significance.",
-    
-    "The Sydney Opera House is a world-renowned performing arts center located in Sydney, Australia. Completed in 1973, the building is famous for its unique sail-like design and stunning location on Sydney Harbour. It is one of the most recognizable architectural landmarks in the world and hosts a wide range of performances, from opera and ballet to theater and concerts. The Opera House is a symbol of Australia’s vibrant cultural scene.",
-    
-    "The Statue of Liberty is a colossal sculpture located on Liberty Island in New York Harbor. A gift from France to the United States in 1886, the statue symbolizes freedom and democracy. Standing at 305 feet tall from base to torch, it was the tallest structure in New York City when it was erected. Today, it remains one of the most famous landmarks in the world and is a symbol of hope and opportunity for millions of immigrants.",
-    
-    "Machu Picchu is an ancient Inca citadel located high in the Andes Mountains of Peru. Built in the 15th century, it was a royal estate and a religious site for the Inca civilization. The site is renowned for its advanced dry-stone construction and stunning location, perched atop a mountain ridge with sweeping views of the surrounding landscape. Machu Picchu is one of the most important archaeological sites in South America and a UNESCO World Heritage site.",
-    
-    "Stonehenge is a prehistoric monument located in Wiltshire, England. Believed to have been constructed between 3000 and 2000 BCE, Stonehenge consists of a ring of standing stones, some weighing up to 25 tons. Its purpose remains a mystery, though it is thought to have been used for religious or ceremonial purposes. Stonehenge attracts thousands of visitors each year who marvel at its ancient mystery and connection to early human history.",
-    
-    "The Acropolis of Athens is an ancient citadel located on a rocky outcrop above the city of Athens, Greece. It is home to several ancient structures, including the Parthenon, a temple dedicated to the goddess Athena. The Acropolis is considered a symbol of classical Greek civilization and is one of the most important archaeological sites in the world. It attracts millions of visitors annually who come to admire its historical and architectural significance.",
-    
-    "The Christ the Redeemer statue is a giant statue of Jesus Christ located in Rio de Janeiro, Brazil. Standing 98 feet tall and with outstretched arms, it overlooks the city from the peak of Corcovado Mountain. Completed in 1931, the statue is an iconic symbol of Christianity and Brazilian culture. It is one of the New Seven Wonders of the World and offers stunning views of Rio and its surrounding beaches.",
+default_documents = []
+
+API_KEY = os.environ.get("MY_SECRET_KEY")
+
+def brave_search(query, count=3):
+    url = "https://api.search.brave.com/res/v1/web/search"
+    headers = {"Accept": "application/json", "X-Subscription-Token": API_KEY}
+    params = {
+        "q": query, 
+        "count": count,
+        "extra_snippets": True
+    }
     
-    "The Kremlin is a fortified complex located in the heart of Moscow, Russia. It serves as the official residence of the Russian president and includes several palaces, cathedrals, and government buildings. The Kremlin has been the seat of Russian power for centuries, from Tsarist Russia to the Soviet Union and modern Russia. It is a UNESCO World Heritage site and a key symbol of Russian political and historical significance.",
+    response = requests.get(url, headers=headers, params=params)
     
-    "The Louvre Museum is one of the world’s largest and most famous art museums, located in Paris, France. Originally a royal palace, it became a museum in 1793 and now houses thousands of works of art, including the famous painting, the Mona Lisa. The Louvre's extensive collection spans from ancient civilizations to modern times. It remains one of the most visited museums in the world, attracting millions of art lovers annually."
-]
+    if response.status_code == 200:
+        results = response.json().get("web", {}).get("results", [])
+        print(results)
+        return [(res["title"], res["description"], res["url"]) for res in results]
+    else:
+        print(f"Error: {response.status_code}, {response.text}")
+        return []
+
 
 # Function to handle the chat with TeapotAI
 def handle_chat(user_input, teapot_ai):
+    results = brave_search(user_input)
+    
+    for i, (title, description, url) in enumerate(results, 1):
+      print(f"{i}. {title}\n   {description.replace('<strong>','').replace('</strong>','')}\n   {url}\n")
+      st.write(f"{i}. {title}\n   {description.replace('<strong>','').replace('</strong>','')}\n   {url}\n")
+
     response = teapot_ai.chat([
         {
             "role": "system",
-            "content": "You are an agent designed to answer facts about famous landmarks."
+            "content": "You are Teapot, an open-source AI assistant optimized for running efficiently on low-end devices. You provide short, accurate responses without hallucinating and excel at extracting information and summarizing text."
         },
         {
             "role": "user",