import pickle
import os



print(os.getcwd())
fileobj=open("/home/user/app/embmmn7.obj","rb")
corpus_embeddings,corpus=pickle.load(fileobj)
fileobj.close()
from sentence_transformers import SentenceTransformer, util
import torch

embedder = SentenceTransformer("ramdane/jurimodel")
import google.generativeai as genai
genai.configure(api_key="AIzaSyCcxB0xY2C1IGDqxlLRmLBH6AX_wbBORX4")

# Set up the model
generation_config = {
  "temperature": 0.9,
  "top_p": 1,
  "top_k": 1,
  "max_output_tokens": 2048,
}

safety_settings = [
  {
    "category": "HARM_CATEGORY_HARASSMENT",
    "threshold": "BLOCK_NONE"
  },
  {
    "category": "HARM_CATEGORY_HATE_SPEECH",
    "threshold": "BLOCK_NONE"
  },
  {
    "category": "HARM_CATEGORY_SEXUALLY_EXPLICIT",
    "threshold": "BLOCK_NONE"
  },
  {
    "category": "HARM_CATEGORY_DANGEROUS_CONTENT",
    "threshold": "BLOCK_NONE"
  },
]

model = genai.GenerativeModel(model_name="gemini-1.0-pro",
                              generation_config=generation_config,
                              safety_settings=safety_settings)
def show(queries):
  query_embedding = embedder.encode(queries, convert_to_tensor=True)
  hits = util.semantic_search(query_embedding, corpus_embeddings, top_k=20)

  hits = hits[0]
  history=[]

    #Get the hits for the first query

  for i in range(0,20):
    history.append({"role": "user", "parts": [corpus[hits[i]['corpus_id']]]})
    history.append({"role": "model", "parts": ["حسنا"]})
  convo = model.start_chat(history=history)
  convo.send_message("       ادا لم يتم طرح سؤال تطلم عن كيفية معالجة الاجتهادات لهده النقط بصفة عامة"+"تخيل انك رجل قانون مختص في القانون الجزائري  تقدم اجابة على اسئلة  قانونية اجب من خلال ما سبق من اجتهادات على السؤال التالي مع دكر الاجتهاد الدي اعتمدت عليه في حالة عدم وجود الجواب في اجتهادات القضائية السابقة اجب على السؤال لكن ادكر انك لم تعتمد على الاجتهاد القضائلا تدكر رقم المواد التي لم تعطى لك في الاجتهادات القضائية"+queries)

  return convo.last.text
import gradio as gr
app = gr.Interface(
  fn=show,
  inputs=gr.Textbox(label="إسئل وسيتم الاجابة عن طريق الاجتهادات القضائية"),
  outputs=gr.TextArea(label="استنتاج النموذج"),
  # Prevents caching conversation history
)

app.launch()