main.py

from fastapi import FastAPI, File, UploadFile
from fastapi.middleware.cors import CORSMiddleware
from faster_whisper import WhisperModel
from duckduckgo_search import DDGS
from openai import OpenAI
import tempfile
import os
import uuid

app = FastAPI()

# Allow mobile app to call backend
app.add_middleware(
    CORSMiddleware,
    allow_origins=["*"],   # dev only
    allow_methods=["*"],
    allow_headers=["*"],
)

# Load Whisper model once
model = WhisperModel("base", compute_type="int8")

# OpenAI client
client = OpenAI(api_key="YOUR_API_KEY")

# Simple in-memory session storage
sessions = {}

@app.post("/search")
async def search(file: UploadFile = File(...)):
    # 1️⃣ Create session
    session_id = str(uuid.uuid4())
    sessions[session_id] = []

    # 2️⃣ Save audio temporarily
    with tempfile.NamedTemporaryFile(delete=False, suffix=".wav") as temp_audio:
        audio_path = temp_audio.name
        temp_audio.write(await file.read())

    try:
        # 3️⃣ Speech → Text
        segments, info = model.transcribe(audio_path, task="translate")
        query = "".join(seg.text for seg in segments).strip()

        # 4️⃣ Web search snippets
        snippets_text = web_search_snippets(query)

        # 5️⃣ Summarize / complete answer with OpenAI GPT-4
        answer = summarize_results(query, snippets_text)

        # 6️⃣ Save memory
        sessions[session_id].append({
            "query": query,
            "answer": answer
        })

        return {
            "session_id": session_id,
            "query": query,
            "result": answer,
            "memory": sessions[session_id]
        }

    finally:
        # 7️⃣ Delete temp audio
        if os.path.exists(audio_path):
            os.remove(audio_path)


# 🔎 Web search - return combined snippets (no truncation)
def web_search_snippets(query: str) -> str:
    results = []
    try:
        with DDGS() as ddgs:
            for r in ddgs.text(query, max_results=5):  # get more results for completeness
                snippet = r.get("snippet")
                if snippet:
                    results.append(snippet)
    except Exception as e:
        print("Web search failed:", e)
        return "I couldn't search the web right now."
    return " ".join(results) if results else "I couldn't find relevant information."


# 🧠 OpenAI GPT-4 summarization & completion
def summarize_results(query: str, results_text: str) -> str:
    prompt = f"""
You are an AI assistant. The following web content may be in any language. 
Please **understand it fully**, summarize it concisely, and answer the question **in English**.

Question: {query}

Web Content:
{results_text}

Answer the question completely and clearly in English:"""
    
    response = client.chat.completions.create(
        model="gpt-4",
        messages=[{"role": "user", "content": prompt}],
        max_tokens=500,
        temperature=0.7
    )
    return response.choices[0].message.content.strip()