Create app.py

app.py

@@ -1,37 +1,26 @@
 import gradio as gr
-from model import MathSolverAI
-from dataset_utils import MathDataset
-from torch.utils.data import DataLoader
+from model_scratch import MathTransformer
+from dataset_utils_scratch import CharTokenizer
+import torch
 
-solver = MathSolverAI()
+# Load trained model (or new untrained)
+tokenizer = CharTokenizer([""])  # initialize empty, replace with your trained tokenizer
+model = MathTransformer(vocab_size=tokenizer.vocab_size).to("cpu")
 
-def solve_question(text_input, image_input):
-    if image_input:
-        answer = solver.solve_image(image_input.name)
-    else:
-        answer = solver.solve_text(text_input)
+def solve_question(text_input):
+    model.eval()
+    with torch.no_grad():
+        x = torch.tensor([tokenizer.encode(text_input)], dtype=torch.long)
+        out = model(x)
+        pred_idx = torch.argmax(out, dim=-1).squeeze().tolist()
+        answer = tokenizer.decode(pred_idx)
     return answer
 
-def train_model(file_input):
-    dataset = MathDataset(file_input.name)
-    loader = DataLoader(dataset, batch_size=2, shuffle=True)
-    solver.train(loader, epochs=1)
-    return "✅ Training completed!"
-
 with gr.Blocks() as demo:
     with gr.Row():
-        with gr.Column(scale=3):
-            text_input = gr.Textbox(label="Enter math question", placeholder="Type your math problem here")
-            image_input = gr.Image(label="Or upload question image", type="filepath")
-            solve_btn = gr.Button("Solve")
-            output = gr.Textbox(label="Solution", interactive=False)
-            solve_btn.click(solve_question, inputs=[text_input, image_input], outputs=output)
-        with gr.Column(scale=1):
-            with gr.Tab("Options"):
-                train_btn = gr.File(label="Upload Dataset to Train", file_types=[".txt", ".json"])
-                train_output = gr.Textbox(label="Training Status")
-                train_btn.change(train_model, inputs=train_btn, outputs=train_output)
-            with gr.Tab("Chat History"):
-                chat_history_box = gr.Textbox(label="Chat history (login with Google to save)", interactive=False)
-                
+        text_input = gr.Textbox(label="Enter Math Question")
+        solve_btn = gr.Button("Solve")
+        output = gr.Textbox(label="Solution")
+        solve_btn.click(solve_question, inputs=text_input, outputs=output)
+
 demo.launch()