Update backend/main.py

backend/main.py

@@ -45,7 +45,10 @@ async def extract_text(file: UploadFile = File(...)):
 @app.post("/summarize")
 async def summarize_text(text: str = Form(...)):
     summary = summarizer(text)
-    return {"summary": summary[0]['summary_text']}
+    if isinstance(summary, list) and len(summary) > 0 and "summary_text" in summary[0]:
+        return {"summary": summary[0]["summary_text"]}
+    else:
+        return {"error": "Summarization failed. No valid response."}
 
 # 3️⃣ تحليل الصور وتوليد وصف لها
 @app.post("/image_caption")
@@ -59,19 +62,29 @@ async def caption_image(file: UploadFile = File(...)):
 
     caption = image_captioning(image_bytes)
     os.remove(file_path)
-    return {"caption": caption[0]['generated_text']}
+
+    if isinstance(caption, list) and len(caption) > 0 and "generated_text" in caption[0]:
+        return {"caption": caption[0]["generated_text"]}
+    else:
+        return {"error": "Image captioning failed. No valid response received."}
 
 # 4️⃣ الإجابة على الأسئلة من النصوص
 @app.post("/qa/")
 async def answer_question(request: QARequest):
     answer = qa_pipeline(request.question, request.context)
-    return {"answer": answer["answer"]}
+    if "answer" in answer:
+        return {"answer": answer["answer"]}
+    else:
+        return {"error": "No valid answer found."}
 
 # 5️⃣ ترجمة المستندات
 @app.post("/translate")
 async def translate_text(text: str = Form(...)):
     translation = translator(text)
-    return {"translation": translation[0]['translation_text']}
+    if isinstance(translation, list) and len(translation) > 0 and "translation_text" in translation[0]:
+        return {"translation": translation[0]["translation_text"]}
+    else:
+        return {"error": "Translation failed. No valid response."}
 
 # 6️⃣ توليد أكواد التصور من أوامر نصية
 @app.post("/generate_code")