Update app.py

app.py

@@ -12,7 +12,7 @@ api = HfApi()
 app = FastAPI() 
 
 modelName = "MitchellKil/gaelic-ipa-byt5" # my fine tuned model
-DATASET_REPO = "mitchellkil/gaelic-app-feedback" # file to write feedback to
+feedbackFile = "mitchellkil/gaelic-app-feedback" # file to write feedback to
 
 HF_TOKEN = os.environ["HF_TOKEN"] # set in secrets
 
@@ -27,11 +27,11 @@ model.eval()
 
 print("Model loaded") 
 
-class TextRequest(BaseModel):
+class gaelicText(BaseModel):
     text: str
 
 
-def text_to_ipa(text: str) -> str:
+def gaelicToIPA(text: str) -> str:
     # Input text - > target text for trained model
     prompt = f"convert Gaelic to IPA: {text}"
 
@@ -52,6 +52,11 @@ def text_to_ipa(text: str) -> str:
     decoded = tokenizer.decode(outputs[0], skip_special_tokens=True)
     return decoded.strip()
 
+@app.post("/predict") # call to actually run and return text to project
+def predict(request: gaelicText):
+    ipa = gaelicToIPA(request.text)
+    return {"ipa": ipa}
+
 class Feedback(BaseModel):
     type: str
     feedback: str
@@ -59,14 +64,10 @@ class Feedback(BaseModel):
     translation: str
     phonetic: str
 
-@app.post("/predict") # call to actually run and return text to project
-def predict(request: TextRequest):
-    ipa_result = text_to_ipa(request.text)
-    return {"ipa": ipa_result}
 
 @app.post("/feedback")
 def save_feedback(data: Feedback):
-    entry = {
+    feedbackText = {
         "type": data.type,
         "timestamp": datetime.utcnow().isoformat(),
         "input_text": data.input_text,
@@ -76,26 +77,27 @@ def save_feedback(data: Feedback):
     }
     try: # gets current feedback file
         local_file = hf_hub_download(
-            repo_id=DATASET_REPO,
+            repo_id=dataset,
             filename="feedback.jsonl",
             repo_type="dataset",
             token=HF_TOKEN
         )
         with open(local_file, "r", encoding="utf-8") as f:
             lines = f.readlines()
-    
+    except Exception:
+        lines = []
 
-    lines.append(json.dumps(entry) + "\n") # append
+    lines.append(json.dumps(feedbackText) + "\n") # append
 
-    temp_path = "feedback_temp.jsonl" # create temp
+    temp_path = "feedbackTemp.jsonl" # create temp file
     with open(temp_path, "w", encoding="utf-8") as f:
         f.writelines(lines)
 
     api.upload_file( # reupload full file
         path_or_fileobj=temp_path,
         path_in_repo="feedback.jsonl",
-        repo_id=DATASET_REPO,
-        repo_type="dataset",
+        repo_id=dataset,
+        repo_type="feedbackFile",
         token=HF_TOKEN,
         commit_message="New feedback added"
     )