Update app.py
Browse files
app.py
CHANGED
|
@@ -149,7 +149,7 @@ def summarize_article(article, content, model, system_prompt, user_query, client
|
|
| 149 |
Article:
|
| 150 |
Title: {article['title']}
|
| 151 |
URL: {article['href']}
|
| 152 |
-
Content: {article['body'][:
|
| 153 |
|
| 154 |
Additional Context:
|
| 155 |
{content[:1000]}... # Truncate additional context as well
|
|
@@ -194,9 +194,12 @@ def get_response_with_search(query, model, num_calls=3, temperature=0.2, use_emb
|
|
| 194 |
search_results = duckduckgo_search(query)
|
| 195 |
client = InferenceClient(model, token=huggingface_token)
|
| 196 |
|
|
|
|
|
|
|
|
|
|
| 197 |
summaries = []
|
| 198 |
for result in search_results:
|
| 199 |
-
summary = summarize_article(result, model, system_prompt, query, client, temperature)
|
| 200 |
summaries.append({
|
| 201 |
"title": result['title'],
|
| 202 |
"url": result['href'],
|
|
|
|
| 149 |
Article:
|
| 150 |
Title: {article['title']}
|
| 151 |
URL: {article['href']}
|
| 152 |
+
Content: {article['body'][:1000]}... # Truncate to avoid extremely long prompts
|
| 153 |
|
| 154 |
Additional Context:
|
| 155 |
{content[:1000]}... # Truncate additional context as well
|
|
|
|
| 194 |
search_results = duckduckgo_search(query)
|
| 195 |
client = InferenceClient(model, token=huggingface_token)
|
| 196 |
|
| 197 |
+
# Prepare overall context
|
| 198 |
+
overall_context = "\n".join([f"{result['title']}\n{result['body']}" for result in search_results])
|
| 199 |
+
|
| 200 |
summaries = []
|
| 201 |
for result in search_results:
|
| 202 |
+
summary = summarize_article(result, overall_context, model, system_prompt, query, client, temperature)
|
| 203 |
summaries.append({
|
| 204 |
"title": result['title'],
|
| 205 |
"url": result['href'],
|