Update app.py

app.py

@@ -1,182 +1,105 @@
-from fastapi import FastAPI, Request
-from fastapi.responses import StreamingResponse
-from fastapi.middleware.cors import CORSMiddleware
-from typing import List, Dict, Any, Optional
-from pydantic import BaseModel
-import asyncio
-import httpx
-
-from config import cookies, headers
-from prompts import ChiplingPrompts
-
-app = FastAPI()
-
-# Add CORS middleware
-app.add_middleware(
-    CORSMiddleware,
-    allow_origins=["*"],
-    allow_credentials=True,
-    allow_methods=["*"],
-    allow_headers=["*"],
-)
-
-# Define request model
-class ChatRequest(BaseModel):
-    message: str
-    messages: List[Dict[Any, Any]]
-    model: Optional[str] = "meta-llama/Llama-4-Maverick-17B-128E-Instruct-FP8"
-
-async def generate(json_data: Dict[str, Any]):
-    max_retries = 5
-    for attempt in range(max_retries):
-        async with httpx.AsyncClient(timeout=None) as client:
-            try:
-                request_ctx = client.stream(
-                    "POST",
-                    "https://api.together.ai/inference",
-                    cookies=cookies,
-                    headers=headers,
-                    json=json_data
-                )
-
-                async with request_ctx as response:
-                    if response.status_code == 200:
-                        async for line in response.aiter_lines():
-                            if line:
-                                yield f"{line}\n"
-                        return
-                    elif response.status_code == 429:
-                        if attempt < max_retries - 1:
-                            await asyncio.sleep(0.5)
-                            continue
-                        yield "data: [Rate limited, max retries]\n\n"
-                        return
-                    else:
-                        yield f"data: [Unexpected status code: {response.status_code}]\n\n"
-                        return
-            except Exception as e:
-                yield f"data: [Connection error: {str(e)}]\n\n"
-                return
-
-    yield "data: [Max retries reached]\n\n"
-
-@app.get("/")
-async def index():
-    return {"status": "ok"}
-
-@app.post("/chat")
-async def chat(request: ChatRequest):
-    current_messages = request.messages.copy()
+from flask import Flask, jsonify
+from flask_cors import CORS
+from googleapiclient.discovery import build
+import random
+
+app = Flask(__name__)
+CORS(app)
+
+API_KEY = 'AIzaSyDO_RMlMevaTD5KHAeY415sKJLqWqrf0mQ'
+youtube = build('youtube', 'v3', developerKey=API_KEY)
+
+def get_video_info(video_id):
+    request = youtube.videos().list(
+        part='snippet,statistics',
+        id=video_id
+    )
+    response = request.execute()
     
-    # Handle both single text or list content
-    if request.messages and isinstance(request.messages[-1].get('content'), list):
-        current_messages = request.messages
-    else:
-        current_messages.append({
-            'content': [{
-                'type': 'text',
-                'text': request.message
-            }],
-            'role': 'user'
-        })
-
-    json_data = {
-        'model': request.model,
-        'max_tokens': None,
-        'temperature': 0.7,
-        'top_p': 0.7,
-        'top_k': 50,
-        'repetition_penalty': 1,
-        'stream_tokens': True,
-        'stop': ['<|eot_id|>', '<|eom_id|>'],
-        'messages': current_messages,
-        'stream': True,
-    }
-
-    return StreamingResponse(generate(json_data), media_type='text/event-stream')
-
-
-@app.post("/generate-modules")
-async def generate_modules(request: Request):
-    data = await request.json()
-    search_query = data.get("searchQuery")
-
-    if not search_query:
-        return {"error": "searchQuery is required"}
-
-    system_prompt = ChiplingPrompts.generateModules(search_query)
-
-    current_messages = [
-        {
-            'role': 'system',
-            'content': [{
-                'type': 'text',
-                'text': system_prompt
-            }]
-        },
-        {
-            'role': 'user',
-            'content': [{
-                'type': 'text',
-                'text': search_query
-            }]
-        }
-    ]
-
-    json_data = {
-        'model': "meta-llama/Llama-4-Maverick-17B-128E-Instruct-FP8",
-        'max_tokens': None,
-        'temperature': 0.7,
-        'top_p': 0.7,
-        'top_k': 50,
-        'repetition_penalty': 1,
-        'stream_tokens': True,
-        'stop': ['<|eot_id|>', '<|eom_id|>'],
-        'messages': current_messages,
-        'stream': True,
+    if not response['items']:
+        return None
+        
+    item = response['items'][0]
+    return {
+        'title': item['snippet']['title'],
+        'views': item['statistics'].get('viewCount', 0),
+        'likes': item['statistics'].get('likeCount', 0),
+        'channel': item['snippet']['channelTitle'],
+        'video_id': item['id'],
+        'thumbnail': item['snippet']['thumbnails']['high']['url']  # Add this line
     }
 
-    return StreamingResponse(generate(json_data), media_type='text/event-stream')
-
-
-@app.post("/generate-topics")
-async def generate_topics(request: Request):
-    data = await request.json()
-    search_query = data.get("searchQuery")
-
-    if not search_query:
-        return {"error": "searchQuery is required"}
-
-    system_prompt = ChiplingPrompts.generateTopics(search_query)
-
-    current_messages = [
-        {
-            'role': 'system',
-            'content': [{
-                'type': 'text',
-                'text': system_prompt
-            }]
-        },
-        {
-            'role': 'user',
-            'content': [{
-                'type': 'text',
-                'text': search_query
-            }]
+def get_trending_videos():
+    request = youtube.videos().list(
+        part='snippet,statistics',
+        chart='mostPopular',
+        regionCode='US',
+        maxResults=50
+    )
+    
+    response = request.execute()
+    
+    videos = []
+    for item in response['items']:
+        video_data = {
+            'title': item['snippet']['title'],
+            'video_id': item['id'],
+            'views': item['statistics'].get('viewCount', 0),
+            'likes': item['statistics'].get('likeCount', 0),
+            'channel': item['snippet']['channelTitle'],
+            'thumbnail': item['snippet']['thumbnails']['high']['url']  # Add this line
         }
-    ]
-
-    json_data = {
-        'model': "meta-llama/Llama-4-Maverick-17B-128E-Instruct-FP8",
-        'max_tokens': None,
-        'temperature': 0.7,
-        'top_p': 0.7,
-        'top_k': 50,
-        'repetition_penalty': 1,
-        'stream_tokens': True,
-        'stop': ['<|eot_id|>', '<|eom_id|>'],
-        'messages': current_messages,
-        'stream': True,
-    }
-
-    return StreamingResponse(generate(json_data), media_type='text/event-stream')
+        videos.append(video_data)
+    return videos
+
+
+@app.route('/random')
+def random_videos():
+    try:
+        videos = get_trending_videos()
+        random_selection = random.sample(videos, min(5, len(videos)))
+        return jsonify(random_selection)
+    except Exception as e:
+        return jsonify({'error': str(e)}), 500
+
+@app.route('/trending')
+def trending_videos():
+    try:
+        videos = get_trending_videos()
+        return jsonify(videos)
+    except Exception as e:
+        return jsonify({'error': str(e)}), 500
+
+@app.route('/info/<video_id>')
+def video_info(video_id):
+    try:
+        info = get_video_info(video_id)
+        if info is None:
+            return jsonify({'error': 'Video not found'}), 404
+        return jsonify(info)
+    except Exception as e:
+        return jsonify({'error': str(e)}), 500
+    
+@app.route('/search/<query>')
+def search_videos(query):
+        try:
+            request = youtube.search().list(
+                part='snippet',
+                q=query,
+                type='video',
+                maxResults=10
+            )
+            response = request.execute()
+            
+            videos = []
+            for item in response['items']:
+                video_id = item['id']['videoId']
+                video_info = get_video_info(video_id)
+                if video_info:
+                    videos.append(video_info)
+            return jsonify(videos)
+        except Exception as e:
+            return jsonify({'error': str(e)}), 500
+
+if __name__ == '__main__':
+    app.run(debug=True)