|
|
from openai import OpenAI
|
|
|
from dotenv import load_dotenv
|
|
|
import os
|
|
|
import requests
|
|
|
import PyPDF2
|
|
|
|
|
|
load_dotenv()
|
|
|
api_key = os.getenv("GEMINI_API_KEY")
|
|
|
model = "gemini-2.5-flash-lite"
|
|
|
base_url = "https://generativelanguage.googleapis.com/v1beta/openai/"
|
|
|
|
|
|
client = OpenAI(base_url=base_url, api_key=api_key)
|
|
|
|
|
|
|
|
|
|
|
|
url = "https://raw.githubusercontent.com/hereandnowai/sathyabama-be-cse-aiml-pt1-07-2025-hands-on-professional-training-on-genai-and-ai-agents/main/general-profile-of-hereandnowai.pdf"
|
|
|
|
|
|
response = requests.get(url)
|
|
|
|
|
|
|
|
|
PDF_File_name = "profile-of-hereandnowai.pdf"
|
|
|
|
|
|
PDF_dir =os.path.dirname(__file__)
|
|
|
PDF_File_path = os.path.join(PDF_dir, PDF_File_name)
|
|
|
with open(PDF_File_path, "wb") as f:
|
|
|
f.write(response.content)
|
|
|
try:
|
|
|
with open(PDF_File_path, "rb") as f:
|
|
|
reader = PyPDF2.PdfReader(f)
|
|
|
pdf_text_chunks = []
|
|
|
for page in reader.pages:
|
|
|
page_text = page.extract_text()
|
|
|
if page_text:
|
|
|
pdf_text_chunks.append(page_text.strip())
|
|
|
pdf_context = "\n".join(pdf_text_chunks) if pdf_text_chunks else "no text found in PDF"
|
|
|
except Exception as e:
|
|
|
print(f"Error reading the PDF file: {e}")
|
|
|
pdf_context = "Error extracting text from PDF file."
|
|
|
|
|
|
|
|
|
|
|
|
def get_response(message, history):
|
|
|
system_prompt = f"""context from {PDF_File_path}:\n{pdf_context}
|
|
|
\n\n Qustion:{message}
|
|
|
\n\n Answer based only on the context:"""
|
|
|
messages = [{"role": "system", "content": system_prompt}]
|
|
|
messages.extend(history)
|
|
|
messages.append({"role": "user", "content": message})
|
|
|
response = client.chat.completions.create(model=model, messages=messages)
|
|
|
ai_response = response.choices[0].message.content
|
|
|
return ai_response
|
|
|
|
|
|
print(get_response("who is the ceo of here and now ai?", [])) |
