File size: 3,186 Bytes
a1dd6d0 2226507 a1dd6d0 2226507 a1dd6d0 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 |
import os
import mimetypes
import cv2
import uuid
import streamlit as st
from google import generativeai as genai
import pywhatkit as pw
# Configure Gemini API
def configure_api():
api_key = "AIzaSyBDkLcFKzE_T6r1E5XjbuXVaoW40Szn71s" # Use your real key here
genai.configure(api_key=api_key)
return genai.GenerativeModel('gemini-2.0-flash')
def capture_from_webcam():
cam = cv2.VideoCapture(0)
if not cam.isOpened():
st.error("Could not open webcam.")
return None
st.info("Capturing image... Press SPACE to capture.")
filename = "ahmad.jpg"
while True:
ret, frame = cam.read()
if not ret:
st.error("Failed to grab frame.")
break
cv2.imshow("Camera Feed - Press SPACE to capture", frame)
key = cv2.waitKey(1)
if key % 256 == 32: # SPACE pressed
cv2.imwrite(filename, frame)
break
elif key % 256 == 27: # ESC
break
cam.release()
cv2.destroyAllWindows()
if os.path.exists(filename):
return filename
return None
def process_image(image_path, model):
mime_type, _ = mimetypes.guess_type(image_path)
if not mime_type:
mime_type = "image/jpeg"
with open(image_path, "rb") as img_file:
image_data = img_file.read()
response = model.generate_content(
[
{"mime_type": mime_type, "data": image_data},
"""complete analyze the given picture pay attention any detail of picture and tell which pakistani song is suitble according person in picture
just give me the name of one hindi song according to peroson in pic and just give the name of song no other detail just name ."""
]
)
return response.text.strip()
# Streamlit UI
st.set_page_config(page_title="Hindi Song Recommender", layout="centered")
st.title("🎶 Hindi Song Recommender from Image")
st.write("### Aap image kaise doge?")
option = st.radio("Choose input method:", ["Upload Image", "Capture from Webcam"])
model = configure_api()
if option == "Upload Image":
uploaded_file = st.file_uploader("Drag and drop your image here", type=["jpg", "jpeg", "png"])
if uploaded_file:
path = f"temp_{uuid.uuid4().hex[:8]}.jpg"
with open(path, "wb") as f:
f.write(uploaded_file.read())
st.session_state.image_path = path
st.image(path, caption="Uploaded Image", use_container_width=True)
elif option == "Capture from Webcam":
if st.button("Capture Image from Webcam"):
path = capture_from_webcam()
if path:
st.session_state.image_path = path
st.image(path, caption="Captured Image", use_container_width=True)
if "image_path" in st.session_state and st.button("Suggest Hindi Song"):
with st.spinner("Analyzing image and finding a song..."):
try:
suggested_song = process_image(st.session_state.image_path, model)
st.success(f"🎵 Suggested Hindi Song: {suggested_song}")
pw.playonyt(suggested_song)
st.info("Playing song on YouTube...")
except Exception as e:
st.error(f"Error: {e}") |