Spaces:
Running
Running
Create app.py
Browse files
app.py
ADDED
@@ -0,0 +1,50 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
1 |
+
import gradio as gr
|
2 |
+
import os
|
3 |
+
import requests
|
4 |
+
|
5 |
+
from langchain_community.document_loaders import WebBaseLoader
|
6 |
+
from langchain_text_splitters import RecursiveCharacterTextSplitter
|
7 |
+
from langchain_community.embeddings import HuggingFaceEmbeddings
|
8 |
+
from langchain_community.vectorstores import FAISS
|
9 |
+
from langchain.chains import RetrievalQA
|
10 |
+
from langchain.chat_models import ChatOpenAI
|
11 |
+
|
12 |
+
# β
Read OpenRouter API key from HF secret
|
13 |
+
OPENROUTER_API_KEY = os.environ.get("ArjunHF")
|
14 |
+
|
15 |
+
class OpenRouterChatModel(ChatOpenAI):
|
16 |
+
def __init__(self, **kwargs):
|
17 |
+
super().__init__(
|
18 |
+
openai_api_base="https://openrouter.ai/api/v1",
|
19 |
+
openai_api_key=OPENROUTER_API_KEY,
|
20 |
+
model_name="mistralai/mistral-small-3.2-24b-instruct:free",
|
21 |
+
**kwargs
|
22 |
+
)
|
23 |
+
|
24 |
+
def qa_on_url(url, question):
|
25 |
+
try:
|
26 |
+
loader = WebBaseLoader(url)
|
27 |
+
docs = loader.load()
|
28 |
+
splitter = RecursiveCharacterTextSplitter(chunk_size=1000, chunk_overlap=100)
|
29 |
+
split_docs = splitter.split_documents(docs)
|
30 |
+
|
31 |
+
embeddings = HuggingFaceEmbeddings(model_name="sentence-transformers/all-MiniLM-L6-v2")
|
32 |
+
vectordb = FAISS.from_documents(split_docs, embeddings)
|
33 |
+
retriever = vectordb.as_retriever()
|
34 |
+
|
35 |
+
llm = OpenRouterChatModel(temperature=0.2)
|
36 |
+
qa_chain = RetrievalQA.from_chain_type(llm, retriever=retriever)
|
37 |
+
return qa_chain.run(question)
|
38 |
+
except Exception as e:
|
39 |
+
return f"β Error: {e}"
|
40 |
+
|
41 |
+
iface = gr.Interface(
|
42 |
+
fn=qa_on_url,
|
43 |
+
inputs=[gr.Textbox(label="Enter Web URL"), gr.Textbox(label="Your Question")],
|
44 |
+
outputs="text",
|
45 |
+
title="π Ask Questions About Any Webpage (Mistral 3.2 via OpenRouter + LangChain)",
|
46 |
+
description="β οΈ This may take 10β20 seconds depending on the page length and LLM response time. Please be patient!"
|
47 |
+
)
|
48 |
+
|
49 |
+
if __name__ == "__main__":
|
50 |
+
iface.launch()
|