vizsum-pro / summarizer.py
Vartex39's picture
summarizer.py dinamik model seçimine göre çalışacak şekilde güncellendi
fe09770
raw
history blame
1.59 kB
import os
import requests
from dotenv import load_dotenv
load_dotenv()
api_key = os.getenv("OPENROUTER_API_KEY")
if not api_key or not api_key.strip():
raise RuntimeError("❌ OPENROUTER_API_KEY bulunamadı. Hugging Face Secrets kısmına eklenmeli.")
def build_prompt(text, mode):
if "Sade" in mode:
instruction = "Bu metni herkesin anlayabileceği şekilde sadeleştir."
elif "Eleştir" in mode:
instruction = "Metni eleştir, eksiklerini ve güçlü yönlerini değerlendir."
elif "Başlık" in mode:
instruction = "Metne uygun birkaç başlık öner."
elif "Not" in mode:
instruction = "Metinden önemli notlar çıkar."
else:
instruction = "Kısa ve teknik bir özet ver."
return f"{instruction}\n\nMetin:\n{text}"
def summarize_text(text, mode, model_name="anthropic/claude-3-haiku"):
url = "https://openrouter.ai/api/v1/chat/completions"
headers = {
"Authorization": f"Bearer {api_key.strip()}",
"Content-Type": "application/json"
}
payload = {
"model": model_name,
"messages": [
{"role": "user", "content": build_prompt(text, mode)}
]
}
try:
response = requests.post(url, headers=headers, json=payload)
response.raise_for_status()
result = response.json()
return result['choices'][0]['message']['content'].strip()
except requests.exceptions.HTTPError as e:
return f"❌ HTTP Hatası: {e} | Yanıt: {response.text}"
except Exception as e:
return f"❌ Sistemsel Hata: {str(e)}"