Spaces:
Sleeping
Sleeping
| # ✅ Gemini-Based Stock Recommendation Extractor (No Audio, No Whisper) | |
| # Uses video metadata (title + description) + Gemini Flash to extract stock info | |
| import gradio as gr | |
| import os | |
| import tempfile | |
| import json | |
| import google.generativeai as genai | |
| from yt_dlp import YoutubeDL | |
| # ✅ Gemini Configuration | |
| GEMINI_MODEL = None | |
| def configure_gemini(api_key): | |
| try: | |
| genai.configure(api_key=api_key) | |
| global GEMINI_MODEL | |
| GEMINI_MODEL = genai.GenerativeModel("gemini-1.5-flash-latest") | |
| return "✅ Gemini API key configured successfully." | |
| except Exception as e: | |
| return f"❌ Gemini configuration failed: {str(e)}" | |
| # ✅ Extract video metadata only (no download) | |
| def extract_metadata(url, cookies_file=None): | |
| try: | |
| ydl_opts = { | |
| 'quiet': True, | |
| 'skip_download': True, | |
| 'noplaylist': True, | |
| } | |
| if cookies_file and os.path.exists(cookies_file): | |
| ydl_opts['cookiefile'] = cookies_file | |
| with YoutubeDL(ydl_opts) as ydl: | |
| info = ydl.extract_info(url, download=False) | |
| return { | |
| 'title': info.get("title", ""), | |
| 'description': info.get("description", ""), | |
| 'duration': info.get("duration", 0), | |
| 'uploader': info.get("uploader", ""), | |
| 'view_count': info.get("view_count", 0), | |
| 'upload_date': info.get("upload_date", "") | |
| }, "✅ Video metadata extracted" | |
| except Exception as e: | |
| return None, f"❌ Metadata extraction failed: {str(e)}" | |
| # ✅ Gemini Prompt for Stock Extraction | |
| def query_gemini_stock_analysis(meta): | |
| if GEMINI_MODEL is None: | |
| return "❌ Gemini model is not initialized." | |
| prompt = f""" | |
| Analyze the following YouTube video metadata and extract any stock trading recommendations: | |
| Title: {meta['title']} | |
| Description: {meta['description']} | |
| Please extract: | |
| - Mentioned companies or stock symbols | |
| - Any price targets, buy/sell/hold recommendations | |
| - Bullish/bearish sentiments if expressed | |
| - If no stock info is present, clearly say "No financial or trading recommendations found." | |
| - Keep the output short and to the point | |
| """ | |
| try: | |
| response = GEMINI_MODEL.generate_content(prompt) | |
| return response.text if response else "⚠️ No response from Gemini." | |
| except Exception as e: | |
| return f"❌ Gemini query failed: {str(e)}" | |
| # ✅ Main Pipeline | |
| def run_pipeline(api_key, url, cookies): | |
| status = configure_gemini(api_key) | |
| if not status.startswith("✅"): | |
| return status, "" | |
| # Save cookies if provided | |
| cookie_path = None | |
| if cookies: | |
| cookie_path = tempfile.mktemp(suffix=".txt") | |
| with open(cookie_path, "wb") as f: | |
| f.write(cookies.read()) | |
| metadata, meta_status = extract_metadata(url, cookie_path) | |
| if not metadata: | |
| return meta_status, "" | |
| result = query_gemini_stock_analysis(metadata) | |
| return meta_status, result | |
| # ✅ Gradio UI | |
| with gr.Blocks(title="Gemini Stock Extractor") as demo: | |
| gr.Markdown(""" | |
| # 📈 Gemini-Based Stock Recommendation Extractor | |
| Paste a YouTube link and get stock-related insights using only the title + description. | |
| No audio, no transcription required. Fast and simple. | |
| """) | |
| with gr.Row(): | |
| api_input = gr.Textbox(label="🔑 Gemini API Key", type="password") | |
| url_input = gr.Textbox(label="YouTube Video URL") | |
| cookies_input = gr.File(label="cookies.txt (optional)", file_types=[".txt"]) | |
| go_btn = gr.Button("🚀 Analyze") | |
| status_box = gr.Textbox(label="Status", lines=1) | |
| output_box = gr.Textbox(label="Extracted Stock Insights", lines=12) | |
| go_btn.click(fn=run_pipeline, inputs=[api_input, url_input, cookies_input], outputs=[status_box, output_box]) | |
| if __name__ == "__main__": | |
| demo.launch(debug=True) | |