vizsum-pro / summarizer.py
Vartex39's picture
VIZSUM PRO REAL DEPLOY — Claude 3, Emoji Yok, PDF Fix
cc21f11
raw
history blame
1.18 kB
import os
import requests
from dotenv import load_dotenv
load_dotenv()
api_key = os.getenv("OPENROUTER_API_KEY").strip()
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):
url = "https://openrouter.ai/api/v1/chat/completions"
headers = {
"Authorization": f"Bearer {api_key}",
"Content-Type": "application/json"
}
data = {
"model": "openai/gpt-3.5-turbo",
"messages": [
{"role": "user", "content": build_prompt(text, mode)}
]
}
response = requests.post(url, headers=headers, json=data)
response.raise_for_status()
return response.json()['choices'][0]['message']['content']