Update app.py

app.py

@@ -3,7 +3,6 @@ from llama_index.core import StorageContext , VectorStoreIndex
 from llama_index.core.base.base_retriever import BaseRetriever
 from llama_index.core.schema import TextNode , NodeRelationship
 from llama_index.embeddings.huggingface_api import HuggingFaceInferenceAPIEmbedding
-from json import loads
 from chromadb import EphemeralClient
 from llama_index.vector_stores.chroma import ChromaVectorStore
 from requests import get as reqget
@@ -11,7 +10,6 @@ from nest_asyncio import apply
 from os import environ
 from pickle import load as pickle_load
 from bs4 import BeautifulSoup
-from io import BytesIO
 import flask
 class AIBook:
   embed_model:HuggingFaceInferenceAPIEmbedding
@@ -49,7 +47,7 @@ book = AIBook()
 def function():
   question = flask.request.args.get("question")
   if question is None:
-    return """Please provide a question as 'https://shethjenil-apiofbookai.hf.space?question=પ્રભુ છે કે નહિ'<br>for get image by book and page https://shethjenil-apiofbookai.hf.space/getbookimage/023657/99<br>for knowing full data with page etc https://shethjenil-apiofbookai.hf.space/fulldetails?question=પ્રભુ છે કે નહિ<br>Change Token as https://shethjenil-apiofbookai.hf.space/changeToken?token=hf_<b>rgPNhjnXpLSodIphwjmRvPbvrovNYnQavj</b><br>Change srcnum as https://shethjenil-apiofbookai.hf.space/changesrcnum?srcnum=2<br><br><form action="/insert_node" method="post" enctype="multipart/form-data"><input type="file" name="file" accept=".book"><input type="submit" value="Upload Book File"></form>"""
+    return """Please provide a question as 'https://shethjenil-apiofbookai.hf.space?question=પ્રભુ છે કે નહિ'<br>for getting image of question https://shethjenil-apiofbookai.hf.space/question_to_img?question=પ્રભુ છે કે નહિ<br>for get image by book and page https://shethjenil-apiofbookai.hf.space/getbookimage/023657/99<br>for knowing full data with page etc https://shethjenil-apiofbookai.hf.space/fulldetails?question=પ્રભુ છે કે નહિ<br>Change Token as https://shethjenil-apiofbookai.hf.space/changeToken?token=hf_<b>rgPNhjnXpLSodIphwjmRvPbvrovNYnQavj</b><br>Change srcnum as https://shethjenil-apiofbookai.hf.space/changesrcnum?srcnum=2<br><br><form action="/insert_node" method="post" enctype="multipart/form-data"><input type="file" name="file" accept=".book"><input type="submit" value="Upload Book File"></form>"""
   if question == "":
     return ""
   return "\n\n".join(AIBook.nodes_to_guj(book.retrieve(AIBook.translate_to_eng(question))))
@@ -76,7 +74,7 @@ def function4():
 def function5():
   question = flask.request.args.get("question")
   if question:
-    return flask.jsonify([{"text":i.get_text(),"score":i.get_score(),"metadata":loads(i.node.to_json())['relationships']['1']['metadata']} for i in book.retrieve(AIBook.translate_to_eng(question))])
+    return flask.jsonify([{"text":i.get_text(),"score":i.get_score(),"metadata":i.node.relationships[NodeRelationship("1")].metadata} for i in book.retrieve(AIBook.translate_to_eng(question))])
   else:
     return "please provide a question parameter https://shethjenil-apiofbookai.hf.space/fulldetails?question=પ્રભુ છે કે નહિ"
 @app.route("/getbookimage/<bookid>/<page>")
@@ -84,6 +82,12 @@ def function6(bookid,page):
 #   response = reqget(BeautifulSoup(reqget(f"https://jainqq.org/explore/{bookid}/{page}").content, "html.parser").find("img",class_="img-fluid").get("src"))
 #   return flask.send_file(BytesIO(response.content), mimetype=response.headers['Content-Type'])
   return flask.redirect(BeautifulSoup(reqget(f"https://jainqq.org/explore/{bookid}/{page}").content, "html.parser").find("img",class_="img-fluid").get("src"))
+@app.route("/question_to_img")
+def function7():
+  question = flask.request.args.get("question")
+  if question:
+    meta = book.retrieve(AIBook.translate_to_eng(question))[0].node.relationships[NodeRelationship("1")].metadata
+    flask.redirect(BeautifulSoup(reqget(f"https://jainqq.org/explore/{meta['bookid']}/{meta['page']}").content, "html.parser").find("img",class_="img-fluid").get("src"))
 
 if __name__ == '__main__':
-    app.run("0.0.0.0",7860)
+    app.run("0.0.0.0",7860)