tons of changes

app.py

@@ -1,132 +1,183 @@
 import os
-import requests
 import gradio as gr
-from openai import OpenAI
-from dotenv import load_dotenv
-# from helpers import format_images, format_videos, format_links, format_local_map, format_knowledge
 from bagoodex_client import BagoodexClient
-
-# Load API key from .env
-load_dotenv()
-AIML_API_KEY = os.getenv('AIML_API_KEY')
-API_URL = 'https://api.aimlapi.com'
+from r_types import ChatMessage
+from prompts import SYSTEM_PROMPT_FOLLOWUP, SYSTEM_PROMPT_MAP, SYSTEM_PROMPT_BASE
+from helpers import (
+    embed_video,
+    embed_image,
+    format_links,
+    embed_google_map,
+    format_knowledge,
+    format_followup_questions
+)
 
 client = BagoodexClient()
 
-def format_knowledge(result):
-    title = result.get('title', 'Unknown')
-    type_ = result.get('type', '')
-    born = result.get('born', '')
-    died = result.get('died', '')
-    content = f"""
-    **{title}**  
-    Type: {type_}  
-    Born: {born}  
-    Died: {died}
+# ----------------------------
+# Chat & Follow-up Functions
+# ----------------------------
+def chat_function(message, history, followup_state, chat_history_state):
     """
-    return gr.Markdown(content)
-
-def format_images(result):
-    urls = [item.get("original", "") for item in result]
-    return urls
-
-# Helper formatting functions
-def format_videos(result):
-    return [vid.get('link', '') for vid in result]
-
-# Advanced search functions
-def perform_video_search(followup_id):
-    if not followup_id:
-        return []
-    result = client.get_videos(followup_id)
-    return format_videos(result)
-
-def format_links(result):
-    links_md = "**Links:**\n"
-    for url in result:
-        title = url.rstrip('/').split('/')[-1]
-        links_md += f"- [{title}]({url})\n"
-    return gr.Markdown(links_md)
-
-# Define the chat function
-def chat_function(message, history, followup_id):
+    Process a new user message.
+    Appends the message and response to the conversation,
+    and retrieves follow-up questions.
+    """
+    # complete_chat returns a new followup id and answer
     followup_id_new, answer = client.complete_chat(message)
-    return answer, followup_id_new
+    # Update conversation history (if history is None, use an empty list)
+    if history is None:
+        history = []
+    updated_history = history + [ChatMessage({"role": "user", "content": message}),
+                                ChatMessage({"role": "assistant", "content": answer})]
+    # Retrieve follow-up questions using the updated conversation
+    followup_questions_raw = client.base_qna(
+        messages=updated_history, system_prompt=SYSTEM_PROMPT_FOLLOWUP
+    )
+    # Format them using the helper
+    followup_md = format_followup_questions(followup_questions_raw)
+    return answer, followup_id_new, updated_history, followup_md
 
-def format_local_map(result):
-    link = result.get('link', '')
-    image_url = result.get('image', '')
-    html = f"""
-    <div>
-        <strong>Local Map:</strong><br>
-        <a href='{link}' target='_blank'>View on Google Maps</a><br>
-        <img src='{image_url}' style='width:100%;'/>
-    </div>
+def handle_followup_click(question, followup_state, chat_history_state):
+    """
+    When a follow-up question is clicked, send it as a new message.
     """
-    return gr.HTML(html)
+    if not question:
+        return chat_history_state, followup_state, ""
+    # Process the follow-up question via complete_chat
+    followup_id_new, answer = client.complete_chat(question)
+    updated_history = chat_history_state + [ChatMessage({"role": "user", "content": question}),
+                                            ChatMessage({"role": "assistant", "content": answer})]
+    # Get new follow-up questions
+    followup_questions_raw = client.base_qna(
+        messages=updated_history, system_prompt=SYSTEM_PROMPT_FOLLOWUP
+    )
+    followup_md = format_followup_questions(followup_questions_raw)
+    return updated_history, followup_id_new, followup_md
 
-def append_local_map(followup_id, chatbot_value):
-    if not followup_id:
-        return chatbot_value
-    result = client.get_local_map(followup_id)
-    formatted = format_local_map(result)
-    new_message = {"role": "assistant", "content": formatted}
-    return chatbot_value + [new_message]
+def handle_local_map_click(followup_state, chat_history_state):
+    """
+    On local map click, try to get a local map.
+    If issues occur, fall back to using the SYSTEM_PROMPT_MAP.
+    """
+    if not followup_state:
+        return chat_history_state
+    try:
+        result = client.get_local_map(followup_state)
+        map_url = result.get('link', '')
+        # Use helper to produce an embedded map iframe
+        html = embed_google_map(map_url)
+    except Exception:
+        # Fall back: use the base_qna call with SYSTEM_PROMPT_MAP
+        result = client.base_qna(
+            messages=chat_history_state, system_prompt=SYSTEM_PROMPT_MAP
+        )
+        # Assume result contains a 'link' field
+        html = embed_google_map(result.get('link', ''))
+    new_message = ChatMessage({"role": "assistant", "content": html})
+    return chat_history_state + [new_message]
 
-def append_knowledge(followup_id, chatbot_value):
-    if not followup_id:
-        return chatbot_value
-    result = client.get_knowledge(followup_id)
-    formatted = format_knowledge(result)
-    new_message = {"role": "assistant", "content": formatted}
-    return chatbot_value + [new_message]
+def handle_knowledge_click(followup_state, chat_history_state):
+    """
+    On knowledge base click, fetch and format knowledge content.
+    """
+    if not followup_state:
+        return chat_history_state
+    result = client.get_knowledge(followup_state)
+    md = format_knowledge(result)
+    new_message = ChatMessage({"role": "assistant", "content": md})
+    return chat_history_state + [new_message]
 
-# Define advanced search functions
-def perform_image_search(followup_id):
-    if not followup_id:
+# ----------------------------
+# Advanced Search Functions
+# ----------------------------
+def perform_image_search(followup_state):
+    if not followup_state:
         return []
-    result = client.get_images(followup_id)
-    urls = format_images(result)
-    return urls
+    result = client.get_images(followup_state)
+    # For images we simply return a list of original URLs
+    return [item.get("original", "") for item in result]
+
+def perform_video_search(followup_state):
+    if not followup_state:
+        return "<p>No followup ID available.</p>"
+    result = client.get_videos(followup_state)
+    # Use the helper to produce the embed iframes (supports multiple videos)
+    return embed_video(result)
 
-def perform_links_search(followup_id):
-    if not followup_id:
+def perform_links_search(followup_state):
+    if not followup_state:
         return gr.Markdown("No followup ID available.")
-    result = client.get_links(followup_id)
+    result = client.get_links(followup_state)
     return format_links(result)
 
-# Custom CSS
+# ----------------------------
+# UI Build
+# ----------------------------
 css = """
 #chatbot {
     height: 100%;
 }
 """
 
-# Build UI
 with gr.Blocks(css=css, fill_height=True) as demo:
+    # State variables to hold followup ID and conversation history, plus follow-up questions text
     followup_state = gr.State(None)
+    chat_history_state = gr.State([])  # holds conversation history as a list of messages
+    followup_md_state = gr.State("")     # holds follow-up questions as Markdown text
+
     with gr.Row():
         with gr.Column(scale=3):
             with gr.Row():
                 btn_local_map = gr.Button("Local Map Search", variant="secondary", size="sm")
                 btn_knowledge = gr.Button("Knowledge Base", variant="secondary", size="sm")
+            # The ChatInterface now uses additional outputs for both followup_state and conversation history,
+            # plus follow-up questions Markdown.
             chat = gr.ChatInterface(
                 fn=chat_function,
                 type="messages",
-                additional_inputs=[followup_state],
-                additional_outputs=[followup_state],
+                additional_inputs=[followup_state, chat_history_state],
+                additional_outputs=[followup_state, chat_history_state, followup_md_state],
             )
-            # Wire up the buttons to append to chat history
+            # Button callbacks to append local map and knowledge base results to chat
             btn_local_map.click(
-                append_local_map,
-                inputs=[followup_state, chat.chatbot],
+                fn=handle_local_map_click,
+                inputs=[followup_state, chat_history_state],
                 outputs=chat.chatbot
             )
             btn_knowledge.click(
-                append_knowledge,
-                inputs=[followup_state, chat.chatbot],
+                fn=handle_knowledge_click,
+                inputs=[followup_state, chat_history_state],
                 outputs=chat.chatbot
             )
+            # Below the chat input, display follow-up questions and let user select one.
+            followup_radio = gr.Radio(
+                choices=[], label="Follow-up Questions (select one and click Send Follow-up)"
+            )
+            btn_send_followup = gr.Button("Send Follow-up")
+            # When a follow-up question is sent, update the chat conversation, followup state, and follow-up list.
+            btn_send_followup.click(
+                fn=handle_followup_click,
+                inputs=[followup_radio, followup_state, chat_history_state],
+                outputs=[chat.chatbot, followup_state, followup_md_state]
+            )
+            # Also display the follow-up questions markdown (for reference) in a Markdown component.
+            followup_markdown = gr.Markdown(label="Follow-up Questions", value="", visible=True)
+            # When the followup_md_state updates, also update the radio choices.
+            def update_followup_radio(md_text):
+                # Assume the helper output is a Markdown string with list items.
+                # We split the text to extract the question lines.
+                lines = md_text.splitlines()
+                questions = []
+                for line in lines:
+                    if line.startswith("- "):
+                        questions.append(line[2:])
+                return gr.update(choices=questions, value=None), md_text
+            followup_md_state.change(
+                fn=update_followup_radio,
+                inputs=[followup_md_state],
+                outputs=[followup_radio, followup_markdown]
+            )
         with gr.Column(scale=1):
             gr.Markdown("### Advanced Search Options")
             with gr.Column(variant="panel"):
@@ -134,21 +185,21 @@ with gr.Blocks(css=css, fill_height=True) as demo:
                 btn_videos = gr.Button("Search Videos")
                 btn_links = gr.Button("Search Links")
                 gallery_output = gr.Gallery(label="Image Results", columns=2)
-                video_output = gr.Gallery(label="Video Results", columns=1, visible=True)
+                video_output = gr.HTML(label="Video Results")  # HTML for embedded video iframes
                 links_output = gr.Markdown(label="Links Results")
                 btn_images.click(
-                    perform_image_search,
+                    fn=perform_image_search,
                     inputs=[followup_state],
                     outputs=[gallery_output]
                 )
                 btn_videos.click(
-                    perform_video_search,
+                    fn=perform_video_search,
                     inputs=[followup_state],
                     outputs=[video_output]
                 )
                 btn_links.click(
-                    perform_links_search,
+                    fn=perform_links_search,
                     inputs=[followup_state],
                     outputs=[links_output]
                 )
-    demo.launch()
+    demo.launch()

bagoodex_client.py

@@ -2,6 +2,9 @@ import os
 import requests
 from openai import OpenAI
 from dotenv import load_dotenv
+from r_types import ChatMessage
+from prompts import SYSTEM_PROMPT_BASE, SYSTEM_PROMPT_MAP
+from typing import List
 
 load_dotenv()
 API_KEY = os.getenv("AIML_API_KEY")
@@ -20,11 +23,24 @@ class BagoodexClient:
         """
         response = self.client.chat.completions.create(
             model="bagoodex/bagoodex-search-v1",
-            messages=[{"role": "user", "content": query}],
+            messages=[
+                ChatMessage(role="user", content=SYSTEM_PROMPT_BASE), 
+                ChatMessage(role="user", content=query)
+            ],
         )
         followup_id = response.id  # the unique ID for follow-up searches
         answer = response.choices[0].message.content
         return followup_id, answer
+    
+    def base_qna(self, messages: List[ChatMessage], system_prompt=SYSTEM_PROMPT_BASE):
+        response = self.client.chat.completions.create(
+            model="gpt-4o",
+            messages=[
+                ChatMessage(role="user", content=system_prompt), 
+                *messages
+            ],
+        )
+        return response.choices[0].message.content
 
     def get_links(self, followup_id):
         headers = {"Authorization": f"Bearer {self.api_key}"}

helpers.py

@@ -1,43 +1,190 @@
-# Helper formatting functions
-def format_local_map(result):
-    html = (
-        f"<div><strong>Local Map:</strong><br>"
-        f"<a href='{result.get('link','')}' target='_blank'>View on Google Maps</a><br>"
-        f"<img src='{result.get('image','')}' style='width:100%;'/></div>"
-    )
-    return html
-
-def format_knowledge(result):
-    html = (
-        f"<div><strong>{result.get('title','Unknown')}</strong><br>"
-        f"Type: {result.get('type','')}<br>"
-        f"Born: {result.get('born','')}<br>"
-        f"Died: {result.get('died','')}</div>"
-    )
-    return html
-
-def format_images(result):
-    # Extract 'original' URL from each image dict.
-    urls = [item.get("original", "") for item in result]
-    return urls
-
-def format_videos(result):
-    html = "<div>"
-    for vid in result:
-        html += (
-            f"<div style='margin-bottom:10px;'>"
-            f"<video controls style='width:100%;' src='{vid.get('link','')}'></video><br>"
-            f"{vid.get('title','')}"
-            f"</div>"
-        )
-    html += "</div>"
-    return html
-
-def format_links(result):
-    html = "<div><ul>"
-    for url in result:
-        # Use the final part of the URL as the title.
-        title = url.rstrip('/').split('/')[-1]
-        html += f"<li><a href='{url}' target='_blank'>{title}</a></li>"
-    html += "</ul></div>"
-    return html
+from dotenv import load_dotenv
+import os
+import gradio as gr
+import urllib.parse
+import re
+from pytube import YouTube
+from typing import List, Optional
+from r_types import (
+    SearchVideosResponse,
+    SearchImagesResponse,
+    SearchLinksResponse,
+    LocalMapResponse,
+    KnowledgeBaseResponse
+)
+
+
+def get_video_id(url: str) -> Optional[str]:
+    """
+    Safely retrieve the YouTube video_id from a given URL using pytube.
+    Returns None if the URL is invalid or an error occurs.
+    """
+    if not url:
+        return None
+
+    try:
+        yt = YouTube(url)
+        return yt.video_id
+    except Exception:
+        # If the URL is invalid or pytube fails, return None
+        return None
+
+
+def embed_video(videos: List[SearchVideosResponse]) -> str:
+    """
+    Given a list of video data (with 'link' and 'title'),
+    returns an HTML string of embedded YouTube iframes.
+    """
+    if not videos:
+        return "<p>No videos found.</p>"
+
+    # Collect each iframe snippet
+    iframes = []
+    for video in videos:
+        url = video.get("link", "")
+        video_id = get_video_id(url)
+        if not video_id:
+            # Skip invalid or non-parsable links
+            continue
+
+        title = video.get("title", "").replace('"', '\\"')  # Escape quotes
+        iframe = f"""
+        <iframe 
+            width="560" 
+            height="315" 
+            src="https://www.youtube.com/embed/{video_id}" 
+            title="{title}" 
+            frameborder="0" 
+            allow="accelerometer; autoplay; clipboard-write; encrypted-media; gyroscope; picture-in-picture" 
+            allowfullscreen>
+        </iframe>
+        """
+        iframes.append(iframe)
+
+    # If no valid videos after processing, return a fallback message
+    if not iframes:
+        return "<p>No valid YouTube videos found.</p>"
+
+    # Join all iframes into one HTML string
+    return "\n".join(iframes)
+
+
+def embed_image(json_data: SearchImagesResponse) -> str:
+    """
+    Given image data with 'original' (URL) and 'title',
+    returns an HTML string with an <img> tag.
+    """
+    title = json_data.get("title", "").replace('"', '\\"')
+    original = json_data.get("original", "")
+
+    if not original:
+        return "<p>No image URL provided.</p>"
+
+    embed_html = f"""
+    <img src="{original}" alt="{title}" style="width:100%">
+    """
+    return embed_html
+
+
+def build_search_links_response(urls: List[str]) -> List[SearchLinksResponse]:
+    """
+    Convert raw URLs into a list of dicts,
+    each with 'title' and 'link' keys for display.
+    """
+    results = []
+    for url in urls:
+        # Extract the last part of the URL as a rough "title"
+        raw_title = url.rstrip("/").split("/")[-1]
+
+        # Decode URL-encoded entities like %20
+        decoded_title = urllib.parse.unquote(raw_title)
+
+        # Replace hyphens/underscores with spaces
+        nice_title = decoded_title.replace("_", " ").replace("-", " ")
+
+        results.append({"title": nice_title, "link": url})
+    return results
+
+
+def format_links(links: List[SearchLinksResponse]) -> str:
+    """
+    Convert a list of {'title': str, 'link': str} objects
+    into a bulleted Markdown string with clickable links.
+    """
+    if not links:
+        return "No links found."
+
+    links_md = "### Links\n\n"
+    for item in links:
+        links_md += f"- [{item['title']}]({item['link']})\n"
+    return links_md
+
+
+def embed_google_map(map_url: str) -> str:
+    """
+    Extracts a textual location from the given Google Maps URL
+    and returns an embedded Google Map iframe for that location.
+    Assumes you have a valid API key in place of 'YOUR_API_KEY'.
+    """
+    load_dotenv()
+    GOOGLE_MAPS_API_KEY = os.getenv("GOOGLE_MAPS_API_KEY")
+
+    if not map_url:
+        return "<p>Invalid Google Maps URL.</p>"
+
+    # Attempt to extract "San+Francisco,+CA" from the URL
+    match = re.search(r"/maps/place/([^/]+)", map_url)
+    if not match:
+        return "Invalid Google Maps URL. Could not extract location."
+
+    location_text = match.group(1)
+    # Remove query params or additional slashes from the captured group
+    location_text = re.split(r"[/?]", location_text)[0]
+
+    # URL-encode location to avoid issues with special characters
+    encoded_location = urllib.parse.quote(location_text, safe="")
+
+    embed_html = f"""
+    <iframe
+      width="600"
+      height="450"
+      style="border:0"
+      loading="lazy"
+      allowfullscreen
+      src="https://www.google.com/maps/embed/v1/place?key={GOOGLE_MAPS_API_KEY}&q={encoded_location}">
+    </iframe>
+    """
+    return embed_html
+
+
+def format_knowledge(result: KnowledgeBaseResponse) -> str:
+    """
+    Given a dictionary of knowledge data (e.g., about a person),
+    produce a Markdown string summarizing that info.
+    """
+    title = result.get("title", "Unknown")
+    type_ = result.get("type", "")
+    born = result.get("born", "")
+    died = result.get("died", "")
+
+    content = f"""
+**{title}**  
+Type: {type_}  
+Born: {born}  
+Died: {died}
+    """
+    return content
+
+
+def format_followup_questions(questions: List[str]) -> str:
+    """
+    Given a list of follow-up questions, return a Markdown string
+    with each question as a bulleted list item.
+    """
+    if not questions:
+        return "No follow-up questions provided."
+
+    questions_md = "### Follow-up Questions\n\n"
+    for question in questions:
+        questions_md += f"- {question}\n"
+    return questions_md

prompts.py

@@ -0,0 +1,46 @@
+SYSTEM_PROMPT_BASE = """
+######SYSTEM INIATED######
+You will be given a conversation chat (e.g., text/ paragraph).
+Answer the given conversation chat with a relevant response.
+
+######NOTE######
+Be nice and polite in your responses!
+######SYSTEM SHUTDOWN######
+"""
+
+SYSTEM_PROMPT_MAP = """
+######SYSTEM INIATED######
+You will be given a content from conversation chat (e.g., text/ paragraph). 
+Your task is to analyze the given content and provide different types of places as close as possible to the given content.
+For exampl: If the given content (conversation chat) was about "How to make a slingshot", you can provide places like "Hardware store", "Woodworking shop", "Outdoor sports store", etc.
+Make sure the places you provide are relevant to the given content. And as much as close to the given content, the better.
+Your final output should be a list of places. 
+Here's JSON format example:
+```json
+{
+  "places": ["Hardware store", "Woodworking shop", "Outdoor sports store"]
+}
+```
+######NOTE######
+Make sure to return only JSON data! Nothing else!
+######SYSTEM SHUTDOWN######
+"""
+
+
+SYSTEM_PROMPT_FOLLOWUP = """
+######SYSTEM INIATED######
+You will be given a content from conversation chat (e.g., text/ paragraph).
+Your task is to analyze the given content and provide a follow-up question based on the given content.
+For example: If the given content (conversation chat) was about "How to make a slingshot", you can provide a follow-up question like "What materials are needed to make a slingshot?".
+Make sure the follow-up question you provide is relevant to the given content.
+Your final output should be a List of follow-up question.
+Here's JSON format example:
+```json
+{
+  "followup_question": ["What materials are needed to make a slingshot?", "How to make a slingshot more powerful?"]
+}
+```
+######NOTE######
+Make sure to return only JSON data! Nothing else!
+######SYSTEM SHUTDOWN######
+"""

r_types.py

@@ -0,0 +1,83 @@
+from typing import TypedDict
+
+# [ChatMessage]:
+# <response>
+# {
+#   "role": "system",
+#   "content": "Hello, how can I help you today?"
+# }
+# </response>
+
+class ChatMessage(TypedDict):
+    role: str
+    content: str
+
+
+# [Search Videos]:
+# <response>
+# Videos:
+# [{'link': 'https://www.youtube.com/watch?v=X9oWGuKypuY', 'thumbnail': 'https://dmwtgq8yidg0m.cloudfront.net/medium/d3G6HeC5BO93-video-thumb.jpeg', 'title': 'Easy Home Made Slingshot'}, {'link': 'https://www.youtube.com/watch?v=V2iZF8oAXHo&pp=ygUMI2d1bGVsaGFuZGxl', 'thumbnail': 'https://dmwtgq8yidg0m.cloudfront.net/medium/sb2Iw9Ug-Pne-video-thumb.jpeg', 'title': 'Making an Apple Wood Slingshot | Woodcraft'}]
+# </response>
+
+class SearchVideosResponse(TypedDict):
+    link: str
+    thumbnail: str
+    title: str
+
+
+# [Search Images]:
+# <response>
+# [{'source': '', 'original': 'https://i.ytimg.com/vi/iYlJirFtYaA/sddefault.jpg', 'title': 'How to make a Slingshot using Pencils ...', 'source_name': 'YouTube'}, {'source': '', 'original': 'https://i.ytimg.com/vi/HWSkVaptzRA/maxresdefault.jpg', 'title': 'How to make a Slingshot at Home - YouTube', 'source_name': 'YouTube'}, {'source': '', 'original': 'https://content.instructables.com/FHB/VGF8/FHXUOJKJ/FHBVGF8FHXUOJKJ.jpg?auto=webp', 'title': 'Country Boy" Style Slingshot ...', 'source_name': 'Instructables'}, {'source': '', 'original': 'https://i.ytimg.com/vi/6wXqlJVw03U/maxresdefault.jpg', 'title': 'Make slingshot using popsicle stick ...', 'source_name': 'YouTube'}, {'source': '', 'original': 'https://ds-tc.prod.pbskids.org/designsquad/diy/DESIGN-SQUAD-42.jpg', 'title': 'Build | Indoor Slingshot . DESIGN SQUAD ...', 'source_name': 'PBS KIDS'}, {'source': '', 'original': 'https://i.ytimg.com/vi/wCxFkPLuNyA/maxresdefault.jpg', 'title': 'Paper Ninja Weapons ...', 'source_name': 'YouTube'}, {'source': '', 'original': 'https://i0.wp.com/makezine.com/wp-content/uploads/2015/01/slingshot1.jpg?fit=800%2C600&ssl=1', 'title': 'Rotating Bearings ...', 'source_name': 'Make Magazine'}, {'source': '', 'original': 'https://makeandtakes.com/wp-content/uploads/IMG_1144-1.jpg', 'title': 'Make a DIY Stick Slingshot Kids Craft', 'source_name': 'Make and Takes'}, {'source': '', 'original': 'https://i.ytimg.com/vi/X9oWGuKypuY/maxresdefault.jpg', 'title': 'Easy Home Made Slingshot - YouTube', 'source_name': 'YouTube'}, {'source': '', 'original': 'https://www.wikihow.com/images/thumb/4/41/Make-a-Sling-Shot-Step-7-Version-5.jpg/550px-nowatermark-Make-a-Sling-Shot-Step-7-Version-5.jpg', 'title': 'How to Make a Sling Shot: 15 Steps ...', 'source_name': 'wikiHow'}]
+# </response>
+
+class SearchImagesResponse(TypedDict):
+    source: str
+    original: str
+    title: str
+    source: str
+    source_name: str
+
+
+# [Links]:
+# <response>
+# ['https://www.reddit.com/r/slingshots/comments/1d50p3e/how_to_build_a_sling_at_home_thats_not_shit/', 'https://www.instructables.com/Make-a-Giant-Slingshot/', 'https://www.mudandbloom.com/blog/stick-slingshot', 'https://pbskids.org/designsquad/build/indoor-slingshot/', 'https://www.instructables.com/How-to-Make-a-Slingshot-2/']
+# </response>
+
+class SearchLinksResponse(TypedDict):
+    title: str
+    link: str
+
+
+### Local Map Response:
+# <response>
+# {
+#   "link": "https://www.google.com/maps/place/San+Francisco,+CA/data=!4m2!3m1!1s0x80859a6d00690021:0x4a501367f076adff?sa=X&ved=2ahUKEwjqg7eNz9KLAxVCFFkFHWSPEeIQ8gF6BAgqEAA&hl=en",
+#   "image": "https://dmwtgq8yidg0m.cloudfront.net/images/TdNFUpcEvvHL-local-map.webp"
+# }
+# </response>
+
+class LocalMapResponse(TypedDict):
+    link: str
+    imgae: str
+
+
+### Model Response:
+
+# <response>
+# ```
+# {
+#   'title': 'Nikola Tesla', 
+#   'type': 'Engineer and futurist', 
+#   'description': None, 
+#   'born': 'July 10, 1856, Smiljan, Croatia', 
+#   'died': 'January 7, 1943 (age 86 years), The New Yorker A Wyndham Hotel, New York, NY'
+# }
+# ```
+# </response>
+
+class KnowledgeBaseResponse(TypedDict):
+    title: str
+    type: str
+    description: str
+    born: str
+    died: str

test_api.py

@@ -0,0 +1,21 @@
+import os
+from openai import OpenAI
+
+client = OpenAI(
+    base_url="https://api.aimlapi.com/v1",
+    api_key=os.getenv("AIML_API_KEY"),
+)
+
+response = client.chat.completions.create(
+    model="gpt-4o",
+    messages=[
+        {
+            "role": "user",
+            "content": "Tell me, why is the sky blue?"
+        },
+    ],
+)
+
+message = response.choices[0].message.content
+
+print(f"Assistant: {message}")

test_app.py

@@ -0,0 +1,41 @@
+import gradio as gr
+import re
+
+def embed_google_map(map_url):
+    """
+    Extracts a textual location from the Google Maps URL and returns
+    an embedded Google Map iframe.
+    """
+    # Simple approach: try to extract "San+Francisco,+CA" from the URL
+    # by capturing what's after '/maps/place/'
+    match = re.search(r'/maps/place/([^/]+)', map_url)
+    if not match:
+        return "Invalid Google Maps URL. Could not extract location."
+
+    location_text = match.group(1)
+    # location_text might contain extra parameters, so let's just
+    # strip everything after the first '?' or '/':
+    location_text = re.split(r'[/?]', location_text)[0]
+
+    embed_html = f"""
+    <iframe
+      width="600"
+      height="450"
+      style="border:0"
+      loading="lazy"
+      allowfullscreen
+      src="https://www.google.com/maps/embed/v1/place?key=YOUR_API_KEY&q={location_text}">
+    </iframe>
+    """
+    return embed_html
+
+with gr.Blocks() as demo:
+    map_url_input = gr.Textbox(label="Enter Google Maps URL")
+    map_output = gr.HTML(label="Embedded Google Map")
+
+    map_url_input.change(embed_google_map, inputs=map_url_input, outputs=map_output)
+
+demo.launch()
+
+
+# <iframe src="https://www.google.com/maps/embed?pb=!1m18!1m12!1m3!1d11988.613116381091!2d69.19850824650604!3d41.3055290002301!2m3!1f0!2f0!3f0!3m2!1i1024!2i768!4f13.1!3m3!1m2!1s0x38ae8bc5b351bc23%3A0x707af898bf0cdb4d!2sEsenin%20St%2018%2C%20Tashkent%2C%20Uzbekistan!5e0!3m2!1sen!2s!4v1740469397909!5m2!1sen!2s" width="600" height="450" style="border:0;" allowfullscreen="" loading="lazy" referrerpolicy="no-referrer-when-downgrade"></iframe>