Update app.py

app.py

@@ -43,7 +43,7 @@ def build_model():
 
 model = build_model()
 
-# If you trained your model:
+# If you trained your model, uncomment:
 # model.load_weights("model.h5")
 
 
@@ -64,12 +64,16 @@ def preprocess_image(image):
 # -------------------
 def decode_predictions(pred):
     input_len = np.ones(pred.shape[0]) * pred.shape[1]
-    results = tf.keras.backend.ctc_decode(pred, input_length=input_len, greedy=True)[0][0]
+
+    results = tf.keras.backend.ctc_decode(
+        pred, input_length=input_len, greedy=True
+    )[0][0]
 
     output_text = ""
     for res in results.numpy()[0]:
         if res != -1:
             output_text += CHARACTERS[res]
+
     return output_text
 
 
@@ -83,15 +87,33 @@ def predict(image):
     return text
 
 
+# -------------------
+# Combined Handler
+# -------------------
+def handle_input(image, typed_text):
+    # If user typed text → return it directly
+    if typed_text is not None and typed_text.strip() != "":
+        return typed_text
+
+    # If image uploaded → run prediction
+    if image is not None:
+        return predict(image)
+
+    return "Please upload an image or enter text."
+
+
 # -------------------
 # Gradio Interface
 # -------------------
 interface = gr.Interface(
-    fn=predict,
-    inputs=gr.Image(type="numpy"),
-    outputs="text",
-    title="Handwritten Line Text Recognition",
-    description="Upload a handwritten line image"
+    fn=handle_input,
+    inputs=[
+        gr.Image(type="numpy", label="Upload Handwritten Image"),
+        gr.Textbox(label="Or Type Text Manually")
+    ],
+    outputs=gr.Textbox(label="Output Text"),
+    title="✍️ Handwritten Line Text Recognition",
+    description="Upload a handwritten line image OR type text manually"
 )
 
 interface.launch()