Update app.py

app.py

@@ -1,52 +1,51 @@
 import os
-from flask import Flask, request, render_template, jsonify
+from fastapi import FastAPI, Request, Form, File, UploadFile
+from fastapi.responses import HTMLResponse, JSONResponse
+from fastapi.templating import Jinja2Templates
 from groq import Groq
+import io
 
 # Set up the Groq client
 os.environ["GROQ_API_KEY"] = "your_groq_api_key_here"
 client = Groq(api_key=os.environ["GROQ_API_KEY"])
 
-app = Flask(__name__)
+# Initialize FastAPI app and template engine
+app = FastAPI()
+templates = Jinja2Templates(directory="templates")
 
-# Route to render the upload form
-@app.route('/')
-def index():
-    return render_template('index.html')
-
-# Route to handle the audio upload and transcription
-@app.route('/transcribe', methods=['POST'])
-def transcribe_audio():
-    audio_data = request.files['audio_data']
-    selected_language = request.form['language']
+@app.get("/", response_class=HTMLResponse)
+async def index(request: Request):
+    return templates.TemplateResponse("index.html", {"request": request})
 
+@app.post("/transcribe")
+async def transcribe_audio(audio_data: UploadFile = File(...), language: str = Form(...)):
     try:
+        audio_content = await audio_data.read()
         # Transcribe the audio based on the selected language
         transcription = client.audio.transcriptions.create(
-            file=(audio_data.filename, audio_data.read()),
+            file=(audio_data.filename, audio_content),
             model="whisper-large-v3",
             prompt="Transcribe the audio accurately based on the selected language.",
             response_format="text",
-            language=selected_language,
+            language=language,
         )
 
-        return jsonify({'transcription': transcription})
-    except Exception as e:
-        return jsonify({'error': str(e)}), 500
+        return JSONResponse(content={'transcription': transcription})
 
-# Route to check grammar and vocabulary of the transcription
-@app.route('/check_grammar', methods=['POST'])
-def check_grammar():
-    transcription = request.form.get('transcription')
-    selected_language = request.form.get('language')
+    except Exception as e:
+        return JSONResponse(status_code=500, content={'error': str(e)})
 
-    if not transcription or not selected_language:
-        return jsonify({'error': 'Missing transcription or language selection'}), 400
+@app.post("/check_grammar")
+async def check_grammar(transcription: str = Form(...), language: str = Form(...)):
+    if not transcription or not language:
+        return JSONResponse(status_code=400, content={'error': 'Missing transcription or language selection'})
 
     try:
         # Grammar check
         grammar_prompt = (
-            f"Briefly check the grammar of the following text in {selected_language}: {transcription}. "
-            "Identify any word that does not belong to the selected language and flag it. Based on the number of incorrect words, also check the grammar deeply and carefully. "
+            f"Briefly check the grammar of the following text in {language}: {transcription}. "
+            "Identify any word that does not belong to the selected language and flag it. "
+            "Based on the number of incorrect words also check the grammar deeply and carefully. "
             "Provide a score from 1 to 10 based on the grammar accuracy, reducing points for incorrect words and make sure to output the score on a new line after two line breaks like 'SCORE='."
         )
 
@@ -58,8 +57,9 @@ def check_grammar():
 
         # Vocabulary check
         vocabulary_prompt = (
-            f"Check the vocabulary accuracy of the following text in {selected_language}: {transcription}. "
-            "Identify any word that does not belong to the selected language and flag it. Based on the number of incorrect words, also check the grammar deeply and carefully. "
+            f"Check the vocabulary accuracy of the following text in {language}: {transcription}. "
+            "Identify any word that does not belong to the selected language and flag it. "
+            "Based on the number of incorrect words also check the grammar deeply and carefully. "
             "Provide a score from 1 to 10 based on the vocabulary accuracy reducing points for incorrect words and make sure to output the score on a new line after two line breaks like 'SCORE='."
         )
 
@@ -69,22 +69,24 @@ def check_grammar():
         )
         vocabulary_feedback = vocabulary_check_response.choices[0].message.content.strip()
 
-        # Return feedback and scores
+        # Calculate scores
         grammar_score = calculate_score(grammar_feedback)
         vocabulary_score = calculate_score(vocabulary_feedback)
 
-        return jsonify({
+        return JSONResponse(content={
             'grammar_feedback': grammar_feedback,
             'vocabulary_feedback': vocabulary_feedback,
             'grammar_score': grammar_score,
             'vocabulary_score': vocabulary_score
         })
+
     except Exception as e:
-        return jsonify({'error': str(e)}), 500
+        return JSONResponse(status_code=500, content={'error': str(e)})
+
 
-def calculate_score(feedback):
+def calculate_score(feedback: str) -> int:
     """ 
-    Calculate score based on feedback content. 
+    Calculate score based on feedback content.
     This function searches for the keyword 'SCORE=' or similar variations 
     (SCORE:, score:, etc.) and extracts the score value.
     """
@@ -92,11 +94,12 @@ def calculate_score(feedback):
     match = re.search(r'(SCORE=|score=|SCORE:|score:|SCORE = )\s*(\d+)', feedback)
     
     if match:
-        # Extract and return the score as an integer
         return int(match.group(2))
     
-    # Return a default score of 0 if no score is found in the feedback
-    return 0
+    return 0  # Return default score of 0 if no score is found
+
 
+# Run the FastAPI app (only needed for local development)
 if __name__ == "__main__":
-    app.run(debug=True)
+    import uvicorn
+    uvicorn.run(app, host="0.0.0.0", port=8000)