Set up api

src/api/main.py

@@ -3,5 +3,4 @@ from routes import homepage_router
 
 app = FastAPI()
 
-# Include the routers in the app
 app.include_router(homepage_router, prefix="/api", tags=["Home Page"])

src/api/routes/home_page_api.py

@@ -3,14 +3,25 @@ import sys
 src_directory = os.path.abspath(os.path.join(os.path.dirname(__file__), "../../..", "src"))
 sys.path.append(src_directory)
 from fastapi import APIRouter, HTTPException
-from schemas import schema
+from schemas.schema import ImageSearch
 from app import homepage
 
-
 router = APIRouter()
 
-@router.post("/{text_querry}", summary="Find the image by text")
-def search_image(search_image: schema.ImageSearch):
-    homepage.get_images_by_text(search_image.querry_text)
+@router.post("/search", summary="Find images by text")
+def search_image(search_request: ImageSearch):
+    try:
+        query = search_request.query_text.strip()
+        if not query:
+            raise HTTPException(status_code=400, detail="Query text cannot be empty")
+
+        images = homepage.get_images_text_query(query)
+        if not images:
+            raise HTTPException(status_code=404, detail="No images found")
+
+        return images
+
+    except Exception as e:
+        raise HTTPException(status_code=500, detail=str(e))
 
     

src/api/schemas/schema.py

@@ -1,5 +1,5 @@
 from pydantic import BaseModel
 
 class ImageSearch(BaseModel):
-    querry_text : str
+    query_text: str = "Dogs" 
 

src/app/app.py

@@ -2,5 +2,6 @@ import homepage
 import torch
 
 homepage.setup_page()
-homepage.search_tab()
+if homepage.get_or_greet_user_name():
+    homepage.search_tab()
 # st.link_button("Navigate to load data page",url="http://localhost:8501/load_data_page")

src/app/homepage.py

@@ -4,7 +4,7 @@ src_directory = os.path.abspath(os.path.join(os.path.dirname(__file__), "../..",
 sys.path.append(src_directory)
 import streamlit as st
 from utils import logger
-from database_pinecone import querry_database,create_database
+from database_pinecone import create_database, query_database
 from model.clip_model import ClipModel
 from data import data_set
 
@@ -25,7 +25,7 @@ def setup_page():
         <i>{PAGE_TITLE} 🔍📸</i>
         </h1>
         """, unsafe_allow_html=True)
-    st.toast("✨ Welcome to Look-a-Like: The Ultimate Image Finder! Start searching now. 🔍")
+    # st.toast("✨ Welcome to Look-a-Like: The Ultimate Image Finder! Start searching now. 🔍")
     logger.info(f"Page successfully configured with title: {PAGE_TITLE}")
 
 def search_tab():
@@ -99,6 +99,7 @@ def display_images(response):
         for i, result in enumerate(response.matches):
             with cols[i % 2]:
                 st.image(result.metadata["url"], width=500)
+                st.write(f"Image score : {result.score}")
         logger.info("Displayed the images successfully")
 
 def write_message(message):
@@ -106,14 +107,14 @@ def write_message(message):
 
 def get_images_by_text(query):
     embedding = clip_model.get_text_embedding(query)
-    response = querry_database.fetch_data(embedding)
+    response = query_database.fetch_data(embedding)
     message = f"🔍 Showing search results for {query}"
     write_message(message)
     images = display_images(response)
 
 def get_images_by_image(query):
     embedding = clip_model.get_uploaded_image_embedding(query)
-    response = querry_database.fetch_data(embedding)
+    response = query_database.fetch_data(embedding)
     message = f"🔍 Showing search results of relevant images"
     write_message(message)
     images = display_images(response)
@@ -161,6 +162,23 @@ def load_data():
         logger.error(f"Error loading data: {e}")
         st.session_state.load_clicked = False
 
+def get_or_greet_user_name():
+    if 'user_name' not in st.session_state:
+        st.session_state.user_name = None
+        logger.info("User_name not found in session_state, setting to None.")
+
+    if st.session_state.user_name is None:
+        logger.info("User_name is None, requesting user input.")
+        user_name = st.text_input("Please let me know your name:",
+                              placeholder="Enter your name buddy")
+        if user_name:
+            st.session_state.user_name = user_name
+            logger.info(f"User entered name: {user_name}. Setting session_state.user_name.")
+            st.rerun()
+    else:
+        logger.info(f"User already entered a name: {st.session_state.user_name}. Displaying greeting.")
+        return st.toast(f"Hello {st.session_state.user_name}! Happy Searching")
+
 
 def about_us():
     if st.button("About us"):

src/app/requests/request_api.py

@@ -0,0 +1,18 @@
+import os
+import sys
+src_directory = os.path.abspath(os.path.join(os.path.dirname(__file__), "../..", "src"))
+sys.path.append(src_directory)
+from utils import logger
+import requests
+
+logger = logger.get_logger()
+
+def get_api(end_point : str = None , query: str = 'Dog'):
+    try:
+        API_URL = f"http://127.0.0.1:8000/api/{end_point}"
+        response = requests.post(API_URL, json={"query_text": query})
+        response.raise_for_status()
+        return response.json()
+    except requests.exceptions.RequestException as e:
+        logger.error(f"API Error: {e}")
+        return []