Update app.py

app.py

@@ -38,10 +38,10 @@ except ImportError:
 import io  # Add io import for DOCX export
 
 # --- Environment variables and constants ---
-FRIENDLI_TOKEN = os.getenv("FRIENDLI_TOKEN", "")
+FIREWORKS_API_KEY = os.getenv("FIREWORKS_API_KEY", "")
 BRAVE_SEARCH_API_KEY = os.getenv("BRAVE_SEARCH_API_KEY", "")
-API_URL = "https://api.friendli.ai/dedicated/v1/chat/completions"
-MODEL_ID = "dep86pjolcjjnv8"
+API_URL = "https://api.fireworks.ai/inference/v1/chat/completions"
+MODEL_ID = "accounts/fireworks/models/qwen3-235b-a22b-instruct-2507"
 DB_PATH = "webnovel_sessions_v1.db"
 
 # Target settings for web novel - UPDATED FOR LONGER EPISODES
@@ -61,9 +61,9 @@ WEBNOVEL_GENRES = {
 }
 
 # --- Environment validation ---
-if not FRIENDLI_TOKEN:
-    logger.error("FRIENDLI_TOKEN not set. Application will not work properly.")
-    FRIENDLI_TOKEN = "dummy_token_for_testing"
+if not FIREWORKS_API_KEY:
+    logger.error("FIREWORKS_API_KEY not set. Application will not work properly.")
+    FIREWORKS_API_KEY = "dummy_token_for_testing"
 
 if not BRAVE_SEARCH_API_KEY:
     logger.warning("BRAVE_SEARCH_API_KEY not set. Web search features will be disabled.")
@@ -379,7 +379,7 @@ class WebNovelDatabase:
 class WebNovelSystem:
     """Web novel generation system"""
     def __init__(self):
-        self.token = FRIENDLI_TOKEN
+        self.api_key = FIREWORKS_API_KEY
         self.api_url = API_URL
         self.model_id = MODEL_ID
         self.tracker = WebNovelTracker()
@@ -387,7 +387,11 @@ class WebNovelSystem:
         WebNovelDatabase.init_db()
 
     def create_headers(self):
-        return {"Authorization": f"Bearer {self.token}", "Content-Type": "application/json"}
+        return {
+            "Accept": "application/json",
+            "Content-Type": "application/json",
+            "Authorization": f"Bearer {self.api_key}"
+        }
 
     # --- Prompt generation functions ---
     def create_planning_prompt(self, query: str, genre: str, language: str) -> str:
@@ -663,10 +667,11 @@ Provide specific improvements."""
                 "model": self.model_id,
                 "messages": full_messages,
                 "max_tokens": max_tokens,
-                "temperature": 0.85,
-                "top_p": 0.95,
-                "presence_penalty": 0.3,
-                "frequency_penalty": 0.3,
+                "temperature": 0.6,
+                "top_p": 1,
+                "top_k": 40,
+                "presence_penalty": 0,
+                "frequency_penalty": 0,
                 "stream": True
             }
             
@@ -1846,6 +1851,7 @@ if __name__ == "__main__":
    
    # Environment check
    logger.info(f"API Endpoint: {API_URL}")
+   logger.info(f"Model: {MODEL_ID}")
    logger.info(f"Target: {TARGET_EPISODES} episodes, {TARGET_WORDS:,} words")
    logger.info("Genres: " + ", ".join(WEBNOVEL_GENRES.keys()))
    
@@ -1862,4 +1868,4 @@ if __name__ == "__main__":
        server_name="0.0.0.0",
        server_port=7860,
        share=False
-   )    
+   )