ya

app/main.py

@@ -5,14 +5,14 @@ from fastapi import FastAPI, File, UploadFile
 from pydantic import BaseModel
 from groq import Groq
 
-
 GROQ_API_KEY = "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"
 groq_client = Groq(api_key=GROQ_API_KEY)
 
 def save_to_file(content: str, filename: str) -> str:
-    with open(filename, 'w', encoding='utf-8') as file:
+    filepath = f"/tmp/{filename}"
+    with open(filepath, 'w', encoding='utf-8') as file:
         file.write(content)
-    return filename
+    return filepath
 
 def transcribe_audio(audio_path: str):
     with open(audio_path, "rb") as audio_file:
@@ -21,22 +21,19 @@ def transcribe_audio(audio_path: str):
             file=audio_file,
             response_format="text"
         )
-    transcription = response
-    return transcription
+    return response
 
 def summarize_soap(dialogue: str):
     prompt_soap = f"""
     Anda adalah asisten medis yang membantu dokter dalam menyusun catatan SOAP berdasarkan percakapan dokter dan pasien.
-    Ringkaskan dalam bentuk paragraf tanpa adanya bullet point dan gunakan bahasa Indonesia.
+    Ringkaskan dalam bentuk paragraf tanpa bullet point dan gunakan bahasa Indonesia.
     Harap buat ringkasan dalam format berikut:
     Subjective:
     Objective:
     Assessment:
     Plan:
-
     ### Percakapan:
     {dialogue}
-
     Tolong jangan tambahkan informasi tambahan selain yang berkaitan dengan diagnosis, obat, hasil lab, dan radiologi.
     """
     response_soap = groq_client.chat.completions.create(
@@ -57,10 +54,8 @@ def detect_medical_tags(dialogue: str):
     Obat:
     Hasil Lab:
     Radiologi:
-
     ### Percakapan:
     {dialogue}
-
     Tolong jangan tambahkan informasi tambahan selain yang berkaitan dengan diagnosis, obat, hasil lab, dan radiologi.
     """
     response_tags = groq_client.chat.completions.create(
@@ -84,26 +79,28 @@ async def root():
 
 @app.post("/full_process")
 async def full_process(audio: UploadFile = File(...)):
-    filename = audio.filename
-    temp_audio_path = f"temp_{filename}"
-    with open(temp_audio_path, "wb") as f:
-        f.write(await audio.read())
-
-    transcription = transcribe_audio(temp_audio_path)
-
-    soap_content, soap_file = generate_soap(transcription)
-    tags_content, tags_file = generate_tags(transcription)
-
-    os.remove(temp_audio_path)  # bersihkan file temporer setelah selesai digunakan
-
-    return {
-        "transcription": transcription,
-        "audio_path": temp_audio_path,
-        "soap_content": soap_content,
-        "soap_file": soap_file,
-        "tags_content": tags_content,
-        "tags_file": tags_file
-    }
+    try:
+        filename = audio.filename
+        temp_audio_path = f"/tmp/temp_{filename}"
+        with open(temp_audio_path, "wb") as f:
+            f.write(await audio.read())
+
+        transcription = transcribe_audio(temp_audio_path)
+
+        soap_content, soap_file = generate_soap(transcription)
+        tags_content, tags_file = generate_tags(transcription)
+
+        os.remove(temp_audio_path)
+
+        return {
+            "transcription": transcription,
+            "soap_content": soap_content,
+            "soap_file": soap_file,
+            "tags_content": tags_content,
+            "tags_file": tags_file
+        }
+    except Exception as e:
+        return {"error": str(e)}
 
 class TranscriptionInput(BaseModel):
     dialogue: str