Update app.py

app.py

@@ -1,6 +1,6 @@
 import os
 import gradio as gr
-from weasyprint import HTML  # For direct HTML to PDF conversion
+from weasyprint import HTML  # Use WeasyPrint for HTML to PDF
 from huggingface_hub import InferenceClient
 from huggingface_hub.utils import HfHubHTTPError
 import requests
@@ -14,7 +14,7 @@ MODEL_NAME = "meta-llama/Llama-3.1-8B-Instruct"
 MAX_INPUT_LENGTH = 4096  # Maximum characters for lesson content
 
 # --- Hugging Face Inference Client ---
-client = InferenceClient(api_key=HF_TOKEN)
+client = InferenceClient(model=MODEL_NAME, token=HF_TOKEN) # Corrected: Use model and token
 
 def call_hf_api(lesson_text: str) -> str:
     """
@@ -179,27 +179,23 @@ Do NOT include any other text outside of the HTML. Output ONLY the HTML.  The HT
     ]
 
     try:
-        # Use the chat completions API correctly
-        response = client.chat_completions.create(
-            model=MODEL_NAME,
-            messages=messages,
-            max_tokens=2000,  # Increased for HTML output
-            temperature=0.7,
-            top_p=0.9,
-            stop=["```"], # Stop at the closing code fence
-        )
-        # Extract the generated content
-        generated_html = response.choices[0].message.content
-
-        # Sometimes LLMs add extra text before or after the HTML.  This cleans it up.
-        if "```html" in generated_html:
-             generated_html = generated_html.split("```html")[1]
-        if "```" in generated_html:
-            generated_html = generated_html.split("```")[0]
+      # Corrected: Use the text-generation task and pass messages directly
+      generated_text = client.text_generation(
+          prompt=prompt,
+          max_new_tokens=2000,
+          temperature=0.7,
+          top_p=0.9,
+          stop_sequences=["```"],
+      )
+
+      # HTML Cleanup:
+      if "```html" in generated_text:
+          generated_text = generated_text.split("```html")[1]
+      if "```" in generated_text:
+          generated_text = generated_text.split("```")[0]
+      return generated_text
 
 
-        return generated_html
-
     except HfHubHTTPError as e:
         return f"Error from Hugging Face Hub: {e}"
     except requests.exceptions.RequestException as e:
@@ -208,6 +204,7 @@ Do NOT include any other text outside of the HTML. Output ONLY the HTML.  The HT
         return f"An unexpected error occurred: {e}"
 
 
+
 def generate_exercises(lesson_text: str) -> str:
     """Generates exercises in HTML format."""
     return call_hf_api(lesson_text)
@@ -245,7 +242,7 @@ with gr.Blocks(title="HSK Exercise Generator", theme=gr.themes.Soft()) as demo:
         )
 
     generate_btn = gr.Button("Generate Exercises")
-    output_html = gr.HTML(label="Generated Exercises")  # Use gr.HTML
+    output_html = gr.HTML(label="Generated Exercises")  # Use gr.HTML for HTML output
 
     with gr.Row():
         download_btn = gr.Button("Download as PDF")