| import re |
| import unicodedata |
| import google.generativeai as genai |
| from fastapi import HTTPException |
| from app.core.config import settings |
|
|
| |
| genai.configure(api_key=settings.GOOGLE_API_KEY) |
| gemini_model = genai.GenerativeModel('gemini-2.0') |
|
|
| def clean_text(story: str) -> str: |
| """Clean and normalize the generated text""" |
| cleaned_story = unicodedata.normalize("NFKD", story) |
| cleaned_story = re.sub(r'[\n\\]+', ' ', cleaned_story) |
| cleaned_story = re.sub(r'\s+', ' ', cleaned_story) |
| cleaned_story = ( |
| cleaned_story.replace("“", "\"").replace("â€", "\"").replace("’", "'") |
| .replace("â€TMs", "'s").replace("“", "\"").replace("Ó", "O") |
| .replace("â", "").replace("�", "") |
| ) |
| cleaned_story = cleaned_story.replace("\\", "") |
| return cleaned_story.strip() |
|
|
| async def refine_story(story: str) -> str: |
| """Refine the story using the Gemini API with two-step refinement""" |
| try: |
| |
| first_prompt = f""" |
| Refine the following story to make it coherent, grammatically correct, and fluent. |
| Maintain the original plot and ending, but enhance readability, fluency, and consistency: |
| |
| {story} |
| """ |
| |
| first_response = gemini_model.generate_content(first_prompt) |
| first_refined = first_response.text if hasattr(first_response, 'text') else str(first_response) |
| |
| |
| final_prompt = f""" |
| Please perform a final polish on this story to ensure perfect coherence, |
| flow, and narrative structure while maintaining the essence of the story: |
| |
| {first_refined} |
| """ |
| |
| final_response = gemini_model.generate_content(final_prompt) |
| final_refined = final_response.text if hasattr(final_response, 'text') else str(final_response) |
| |
| return final_refined if final_refined else "No refined story generated." |
| |
| except Exception as e: |
| print(f"Error in story refinement: {str(e)}") |
| return story |
