linting

app.py

@@ -1,18 +1,24 @@
-import gradio as gr
 import json
-from datetime import datetime
-from theme import TufteInspired
-import uuid
-from huggingface_hub import InferenceClient, CommitScheduler, hf_hub_download
-from openai import OpenAI
-from huggingface_hub import get_token, login
-from prompts import detailed_genre_description_prompt, basic_prompt, very_basic_prompt
-import random
 import os
-from pathlib import Path
 import random
+import uuid
+from datetime import datetime
+from pathlib import Path
 from typing import Callable, List, Tuple
 
+import gradio as gr
+from huggingface_hub import (
+    CommitScheduler,
+    InferenceClient,
+    get_token,
+    hf_hub_download,
+    login,
+)
+from openai import OpenAI
+
+from prompts import basic_prompt, detailed_genre_description_prompt, very_basic_prompt
+from theme import TufteInspired
+
 HF_TOKEN = os.getenv("HF_TOKEN")
 
 # Define available models
@@ -96,7 +102,6 @@ def create_client(model_id):
 # )
 
 
-
 def weighted_random_choice(choices: List[Tuple[Callable, float]]) -> Callable:
     total = sum(weight for _, weight in choices)
     r = random.uniform(0, total)
@@ -107,16 +112,18 @@ def weighted_random_choice(choices: List[Tuple[Callable, float]]) -> Callable:
         upto += weight
     assert False, "Shouldn't get here"
 
+
 def generate_prompt() -> str:
     prompt_choices = [
         (detailed_genre_description_prompt, 0.5),
         (basic_prompt, 0.3),
         (very_basic_prompt, 0.2),
     ]
-    
+
     selected_prompt_func = weighted_random_choice(prompt_choices)
     return selected_prompt_func()
 
+
 def get_and_store_prompt():
     prompt = generate_prompt()
     print(prompt)  # Keep this for debugging

genere_descriptions.py

@@ -48,7 +48,7 @@ very_short_descriptions = [
     "Cli-Fi",
     "Weird Fiction",
     "New Adult",
-    "Afrofuturism"
+    "Afrofuturism",
 ]
 
 short_descriptions = [
@@ -101,7 +101,7 @@ short_descriptions = [
     "Satirical fantasy critiquing modern society through a fantastical lens",
     "Hardboiled noir fiction set in 1940s Los Angeles",
     "Speculative fiction exploring the ethical implications of biotechnology",
-    "Magical realism rooted in Native American folklore and traditions"
+    "Magical realism rooted in Native American folklore and traditions",
 ]
 
 
@@ -162,8 +162,7 @@ In essence, this genre offers a sober, realistic, and often bleak portrayal of t
     "This genre explores the intersection of crime, politics, and social issues in urban settings. Stories typically follow detectives, journalists, or ordinary citizens as they navigate complex webs of corruption and violence in city environments. The writing style is often gritty and realistic, with sharp dialogue and vivid descriptions of urban landscapes. Themes revolve around systemic injustice, the corrupting influence of power, and the moral ambiguities faced by those seeking truth in a compromised world. These novels offer a critique of societal institutions while delivering tense, plot-driven narratives.",
     "Blending elements of horror with more traditional fantasy settings, this genre creates dark, atmospheric worlds where magic comes at a terrible price. Stories often feature morally ambiguous protagonists struggling against cosmic horrors, ancient evils, or the corrupting influence of dark powers. The writing style tends to be descriptive and evocative, building a sense of dread and unease. Themes explore the nature of evil, the corrupting influence of power, and the psychological toll of confronting the unknown. These novels challenge the traditionally heroic narratives of fantasy, offering a more pessimistic, horror-tinged view of magical worlds.",
     "Set in the vast expanse of space or on distant planets, this genre explores the challenges and wonders of interstellar travel and colonization. Stories typically feature advanced technologies, encounters with alien species, and the struggle for survival in hostile environments. The writing style often balances technical details with grand, imaginative vistas and complex character dynamics. Themes revolve around human adaptability, the ethics of colonization, and the philosophical implications of humanity's place in a vast, often indifferent universe. These novels invite readers to contemplate the future of human civilization and the potential wonders and terrors that await us among the stars.",
-, """This genre of science fiction often focuses on hard science and technological speculation, blending rigorous scientific concepts with imaginative storytelling. The works typically feature:
-
+    """ This genre of science fiction often focuses on hard science and technological speculation, blending rigorous scientific concepts with imaginative storytelling. The works typically feature:
 Detailed exploration of scientific ideas and their potential consequences
 Emphasis on problem-solving and intellectual challenges
 Characters who are scientists, engineers, or technically-minded individuals
@@ -172,4 +171,5 @@ Realistic depictions of space travel and extraterrestrial environments
 Alternative interpretations of historical events or scientific theories
 Critique of established scientific paradigms and institutions
 
-The stories often challenge readers to think critically about scientific concepts and their implications for society. While maintaining scientific plausibility, the author often introduces unconventional ideas that push the boundaries of accepted theories, encouraging readers to question established norms and consider alternative perspectives on the nature of reality and humanity's place in the universe."""]
+The stories often challenge readers to think critically about scientific concepts and their implications for society. While maintaining scientific plausibility, the author often introduces unconventional ideas that push the boundaries of accepted theories, encouraging readers to question established norms and consider alternative perspectives on the nature of reality and humanity's place in the universe.""",
+]

prompts.py

@@ -1,6 +1,12 @@
-from genere_descriptions import short_descriptions, detailed_descriptions, very_short_descriptions
 import random
 
+from genere_descriptions import (
+    detailed_descriptions,
+    short_descriptions,
+    very_short_descriptions,
+)
+
+
 def very_basic_prompt():
     genre = random.choice(very_short_descriptions)
     return f"""Write a blurb for a book in the following genre {genre}