review_api.py

import os
import requests
import random
import logging
import json

try:
    from PyPDF2 import PdfReader
except ImportError:
    # Fallback for Docker environment
    from pypdf import PdfReader

# Set up logging
logging.basicConfig(level=logging.INFO)
logger = logging.getLogger(__name__)

# Check for environment variable or use a default value for testing
def get_api_key():
    api_key = os.environ.get('DEEPSEEK_API_KEY')
    if not api_key:
        logger.warning("No DEEPSEEK_API_KEY environment variable found. Using mock responses.")
    return api_key

# Mock evaluation result for testing
def get_mock_evaluation(story_excerpt):
    # Use a short excerpt of the story in the evaluation to make it look personalized
    story_start = story_excerpt[:100].replace("\n", " ").strip()
    if len(story_start) > 50:
        story_start = story_start[:50] + "..."

    # Generate random scores but keep them reasonable
    unity_score = random.randint(6, 9)
    characters_score = random.randint(6, 9)
    decisive_moment_score = random.randint(7, 10)
    language_score = random.randint(5, 9)
    time_place_score = random.randint(6, 9)
    plot_score = random.randint(7, 10)
    ending_score = random.randint(6, 10)
    message_score = random.randint(6, 9)
    
    total_score = unity_score + characters_score + decisive_moment_score + language_score + \
                 time_place_score + plot_score + ending_score + message_score
    
    # Mock evaluation text
    return {
        "evaluation": f"""📋 التقييم:



وحدة الحدث: {unity_score}/10

القصة تدور حول حدث رئيسي بشكل جيد، وتحافظ على تماسك الأحداث.



الشخصيات المحدودة والمعرّفة: {characters_score}/10

الشخصيات محددة بوضوح ومتميزة، مع تطوير مناسب للشخصيات الرئيسية.



التركيز على لحظة حاسمة: {decisive_moment_score}/10

القصة تبني بمهارة نحو لحظة التحول الرئيسية التي تغير مسار الأحداث.



الإيجاز واقتصاد اللغة: {language_score}/10

اللغة موجزة ومعبرة، تنقل المعاني بفعالية دون إطناب غير ضروري.



وحدة الزمان والمكان: {time_place_score}/10

هناك استخدام متناسق للإطار الزماني والمكاني، مما يعزز تماسك القصة.



حبكة جيدة البناء: {plot_score}/10

الحبكة متطورة بشكل منطقي ومتماسك، مع تسلسل واضح للأحداث.



نهاية مؤثرة: {ending_score}/10

النهاية تترك انطباعًا قويًا وتوفر إغلاقًا مناسبًا للأحداث.



رسالة أو موضوع واضح: {message_score}/10

تُظهر القصة رسالة واضحة تتطور بشكل طبيعي من خلال الأحداث.



النتيجة النهائية: {total_score}/80

{"هذا عمل أدبي متميز يتسم بقوة البناء والتماسك. القصة تحقق توازنًا جيدًا بين تطوير الشخصيات وتقدم الحبكة." if total_score > 65 else "رغم وجود عناصر قوية في القصة، هناك مجال للتحسين في بعض الجوانب. ننصح بالتركيز على تعزيز تماسك الأحداث وتطوير الشخصيات بشكل أعمق."}



ملاحظة: بداية القصة "{story_start}" تظهر إمكانات جيدة وتجذب اهتمام القارئ.""",
        "fixed_story": None
    }

def review_story(pdf_path):
    """

    Review a story from a PDF file

    """
    try:
        # Load text from PDF
        text = ""
        try:
            with open(pdf_path, "rb") as f:
                reader = PdfReader(f)
                text = "\n".join([p.extract_text() for p in reader.pages if p.extract_text()])
            text = " ".join(text.split())
        except Exception as e:
            logger.error(f"PDF loading failed: {e}")
            return {"evaluation": f"Error loading PDF: {e}", "fixed_story": None}
        
        # Get evaluation
        api_key = get_api_key()
        
        if not api_key:
            # Return mock evaluation
            logger.info("Using mock evaluation for PDF")
            return get_mock_evaluation(text)
            
        # Use API for evaluation
        return call_api_for_evaluation(text)
            
    except Exception as e:
        logger.error(f"Error in review_story: {e}")
        return {"evaluation": f"Error processing story: {e}", "fixed_story": None}

def review_story_text(story_text):
    """

    Review a story provided as text directly

    """
    try:
        # Get evaluation
        api_key = get_api_key()
        
        if not api_key:
            # Return mock evaluation
            logger.info("Using mock evaluation for text")
            return get_mock_evaluation(story_text)
            
        # Use API for evaluation
        return call_api_for_evaluation(story_text)
            
    except Exception as e:
        logger.error(f"Error in review_story_text: {e}")
        return {"evaluation": f"Error processing story text: {e}", "fixed_story": None}

def call_api_for_evaluation(story):
    """

    Call the DeepSeek API for story evaluation

    """
    api_key = get_api_key()
    if not api_key:
        return get_mock_evaluation(story)
        
    evaluation_prompt = f"""

    You are a professional literary critic specializing in the art of the short story. Your task is to evaluate the following story according to 8 essential criteria used in literary criticism. You must write the full evaluation in Modern Standard Arabic, using a clear and organized style.



    🔹 For each criterion:

    Give a score out of 10.

    Write a brief explanation (one or two lines) that justifies the score.



    🔹 At the end of the evaluation:

    Add up the scores to get a final result out of 80.

    If the score is above 65: Praise the story as a successful and well-crafted literary work.

    If the score is 65 or lower: Provide constructive criticism that highlights the main weaknesses and suggests how to improve them.



    Start the evaluation with a title that includes an emoji, such as: 📋 التقييم:

    

    🔹 The evaluation criteria are:

    1. Unity of event: Does the story revolve around a single main incident or situation?

    2. Limited and defined characters: Does the story include a small number of clear and distinctive characters?

    3. Focus on a decisive moment: Does the story highlight a turning point or critical decision?

    4. Conciseness and economy of language: Is the language focused and free of unnecessary details?

    5. Unity of time and place: Does the story take place in a specific time and setting?

    6. Well-structured plot: Is there a clear logical sequence (beginning, middle, end)?

    7. Impactful ending: Does the ending leave an emotional or intellectual impact?

    8. Clear message or theme: Does the story convey a specific idea or feeling clearly?



    Evaluate the following story based on these criteria:

    {story}

    """

    url = "https://api.deepseek.com/v1/chat/completions"
    headers = {
        "Authorization": f"Bearer {api_key}",
        "Content-Type": "application/json"
    }

    payload_eval = {
        "model": "deepseek-chat",
        "messages": [
            {"role": "system", "content": "You are a formal short story evaluator."},
            {"role": "user", "content": evaluation_prompt.strip()}
        ],
        "temperature": random.uniform(0.9, 1.0),
        "max_tokens": 2500
    }

    try:
        logger.info("Sending request to DeepSeek API")
        response_eval = requests.post(url, headers=headers, json=payload_eval)
        
        if response_eval.status_code != 200:
            logger.error(f"API Error: {response_eval.status_code} - {response_eval.text}")
            return {"evaluation": f"Error from API: {response_eval.text}", "fixed_story": None}

        evaluation_result = response_eval.json()["choices"][0]["message"]["content"]
        logger.info("Successfully received evaluation from API")

        return {
            "evaluation": evaluation_result.strip(),
            "fixed_story": None
        }
    except Exception as e:
        logger.error(f"API call failed: {e}")
        return {"evaluation": f"Error calling API: {e}", "fixed_story": None}