Update app.py

app.py

@@ -1,13 +1,16 @@
 import gradio as gr
-import requests
 import os
 from huggingface_hub import hf_hub_download
-from handwriting_api import InputData
-from typing import List
 
+# 1. Import your model logic directly, not via HTTP
+from handwriting_api import InputData, validate_input
+from hand import Hand
 
-# Hugging Face Spaces requires app to run on port 7860
-GRADIO_PORT = int(os.getenv("GRADIO_SERVER_PORT", "7860"))
+
+# 2. Initialize your model once
+hand = Hand()
+
+# 3. Directly call the model logic in the Gradio function
 def generate_handwriting_interface(text: str, style: int) -> str:
     data = InputData(
         text=text,
@@ -17,44 +20,44 @@ def generate_handwriting_interface(text: str, style: int) -> str:
         stroke_widths=[2]
     )
     
-    response = requests.post(
-        "http://localhost:8000/synthesize",
-        json=data.dict()
+    # Validate the input
+    try:
+        validate_input(data)
+    except ValueError as e:
+        return f"Error: {str(e)}"
+
+    # Generate the handwriting
+    lines = data.text.split('\n')
+    biases = [data.bias] * len(lines)
+    styles = [data.style] * len(lines)
+
+    hand.write(
+        filename='img/output.svg',
+        lines=lines,
+        biases=biases,
+        styles=styles,
+        stroke_colors=data.stroke_colors,
+        stroke_widths=data.stroke_widths
     )
-    
-    if response.status_code == 200:
-        return open("img/output.svg").read()
-    else:
-        return f"Error: {response.json()['detail']}"
 
+    # Return the SVG contents
+    with open("img/output.svg", "r") as f:
+        return f.read()
+
+# 4. Set up the Gradio UI
 with gr.Blocks() as demo:
     gr.Markdown("# Handwriting Synthesis")
     with gr.Row():
-        text_input = gr.Textbox(label="Input Text", lines=3, placeholder="Enter text to convert to handwriting...")
+        text_input = gr.Textbox(label="Input Text", lines=3)
         style_select = gr.Slider(minimum=0, maximum=12, step=1, label="Style", value=0)
     
     output_svg = gr.HTML(label="Generated Handwriting")
-    download_button = gr.Button("Download SVG")
-    
-    inputs = [text_input, style_select]
-    outputs = [output_svg]
-    
     generate_button = gr.Button("Generate")
+
     generate_button.click(
         fn=generate_handwriting_interface,
-        inputs=inputs,
-        outputs=outputs
-    )
-    
-    download_button.click(
-        fn=lambda svg: svg,
-        inputs=output_svg,
-        outputs=gr.File(label="Download SVG")
+        inputs=[text_input, style_select],
+        outputs=output_svg
     )
 
-demo.launch(
-    server_name="0.0.0.0",
-    server_port=GRADIO_PORT,
-    share=False,
-    show_error=True
-)
+demo.launch(server_name="0.0.0.0", server_port=7860)