sachin commited on
Commit
53ace05
·
1 Parent(s): 31bd7d3

update-call

Browse files
Files changed (1) hide show
  1. src/server/main.py +18 -4
src/server/main.py CHANGED
@@ -688,6 +688,9 @@ async def translate(
688
  }
689
 
690
  try:
 
 
 
691
  response = requests.post(
692
  external_url,
693
  json=payload,
@@ -700,10 +703,11 @@ async def translate(
700
  response.raise_for_status()
701
 
702
  response_data = response.json()
703
- translations = response_data.get("translations", [])
 
704
 
705
  if not translations or len(translations) != len(decrypted_sentences):
706
- logger.warning(f"Unexpected response format: {response_data}")
707
  raise HTTPException(status_code=500, detail="Invalid response from translation service")
708
 
709
  logger.debug(f"Translation successful: {translations}")
@@ -892,6 +896,15 @@ async def visual_query(
892
  files = {"file": (file.filename, decrypted_content, file.content_type)}
893
  data = {"query": decrypted_query}
894
 
 
 
 
 
 
 
 
 
 
895
  response = requests.post(
896
  external_url,
897
  files=files,
@@ -902,10 +915,11 @@ async def visual_query(
902
  response.raise_for_status()
903
 
904
  response_data = response.json()
905
- answer = response_data.get("answer", "")
 
906
 
907
  if not answer:
908
- logger.warning(f"Empty answer received from external API: {response_data}")
909
  raise HTTPException(status_code=500, detail="No answer provided by visual query service")
910
 
911
  logger.debug(f"Visual query successful: {answer}")
 
688
  }
689
 
690
  try:
691
+ response = dwani.Translate.run_translate(sentences=decrypted_sentences, src_lang=decrypted_src_lang, tgt_lang=decrypted_tgt_lang)
692
+
693
+ '''
694
  response = requests.post(
695
  external_url,
696
  json=payload,
 
703
  response.raise_for_status()
704
 
705
  response_data = response.json()
706
+ '''
707
+ translations = response.get("translations", [])
708
 
709
  if not translations or len(translations) != len(decrypted_sentences):
710
+ logger.warning(f"Unexpected response format: {response}")
711
  raise HTTPException(status_code=500, detail="Invalid response from translation service")
712
 
713
  logger.debug(f"Translation successful: {translations}")
 
896
  files = {"file": (file.filename, decrypted_content, file.content_type)}
897
  data = {"query": decrypted_query}
898
 
899
+
900
+
901
+ response = dwani.Vision.caption(
902
+ file_path=files,
903
+ query=decrypted_query,
904
+ src_lang=decrypted_src_lang,
905
+ tgt_lang=decrypted_tgt_lang
906
+ )
907
+ '''
908
  response = requests.post(
909
  external_url,
910
  files=files,
 
915
  response.raise_for_status()
916
 
917
  response_data = response.json()
918
+ '''
919
+ answer = response.get("answer", "")
920
 
921
  if not answer:
922
+ logger.warning(f"Empty answer received from external API: {response}")
923
  raise HTTPException(status_code=500, detail="No answer provided by visual query service")
924
 
925
  logger.debug(f"Visual query successful: {answer}")