OPENAI - REFERENCE RAG

Simon-Pierre Boucher

2024-09-14

import requests
import json
from scholarly import scholarly  # Vous aurez besoin d'installer la bibliothèque scholarly: pip install scholarly
import os
import re
from dotenv import load_dotenv

# Load environment variables from .env file
load_dotenv()

# Access the API key from environment variables
api_key = os.getenv("OPENAI_API_KEY")

def generate_text_with_references(prompt, num_references):
    """
    Recherche des références académiques et génère un texte en utilisant OpenAI.

    Paramètres:
    prompt (str): Le texte initial pour lequel générer un rapport.
    num_references (int): Le nombre de références à récupérer.

    Retourne:
    str: Texte généré par l'API OpenAI avec références.
    """
    
    # Recherche des références académiques sur Google Scholar
    search_query = scholarly.search_pubs(prompt)
    references = []
    for i in range(num_references):
        try:
            pub = next(search_query)
            references.append(f"{i+1}. {pub['bib']['title']} - {pub['bib']['author']} ({pub['bib']['pub_year']})")
        except StopIteration:
            break

    # Formater les références
    formatted_references = "\n".join(references)
    
    # Générer le contenu du prompt avec les références
    prompt_content = f"""
    {prompt}

    Références académiques :
    {formatted_references}

    Veuillez générer un texte détaillé et informatif en utilisant les informations ci-dessus et en incluant les références académiques fournies.
    """

    # Configurer les en-têtes de la requête
    headers = {
        "Content-Type": "application/json",
        "Authorization": f"Bearer {api_key}"
    }

    # Configurer les données de la requête
    data = {
        "model": "gpt-4",
        "messages": [
            {"role": "user", "content": prompt_content}
        ],
        "temperature": 0.7
    }

    # Envoyer la requête à l'API OpenAI
    response = requests.post("https://api.openai.com/v1/chat/completions", headers=headers, data=json.dumps(data))

    # Traiter la réponse
    if response.status_code == 200:
        response_json = response.json()
        generated_text = response_json["choices"][0]["message"]["content"].strip()
        
        # Ajouter les références à la fin du texte généré
        final_text = f"{generated_text}\n\nRéférences:\n{formatted_references}"
        return final_text
    else:
        return f"Erreur {response.status_code}: {response.text}"

# Exemple d'utilisation
prompt = "Impact of COVID-19 on world economy"  # Remplacez par le prompt que vous voulez interroger
num_references = 10  # Spécifiez le nombre de références que vous voulez obtenir
generated_text = generate_text_with_references(prompt, num_references)
print("\nTexte généré avec références :")
print(generated_text)

Texte généré avec références :
The COVID-19 pandemic has had a profound impact on the global economy, leading to unprecedented financial and social consequences. Many researchers have conducted extensive studies on this subject, shedding light on various aspects of the crisis and its implications for the world economy.

In their paper, Bagchi, Chatterjee, Ghosh, and Dandapat (2020) discuss the immediate impact of the pandemic on the global economy. The authors highlight the drastic reduction in global productivity due to lockdown measures, the surge in unemployment rates, and the collapse of small and medium-sized enterprises. They argue that the pandemic has exacerbated economic disparities, particularly in developing countries.

Similarly, Naseer, Khalid, Parveen, and Abbass (2023) shed light on the economic fallout from the COVID-19 outbreak. They point out the disruption of global supply chains and the resulting impact on international trade. The authors also discuss the strain on healthcare systems worldwide and the economic implications of such pressure.

Mishra (2020) takes a more future-oriented perspective, discussing the post-pandemic world and its impact on the global economy. He suggests that the crisis may lead to a reevaluation of globalization as countries become more aware of their vulnerabilities in the face of global disruptions.

The effects of the pandemic on energy and environment have also been explored. Priya, Cuce, and Sudhakar (2021) consider the pandemic's impact on global energy consumption and environmental pollution. They observe a significant decrease in global energy demand and a corresponding drop in pollution levels during lockdown periods.

In a broader perspective, Shrestha, Shad, Ulvi, and Khan (2020) discuss the impact of COVID-19 on globalization. They argue that the pandemic has revealed the fragility of global interconnectivity, leading to a potential shift towards a more national-focused approach in various sectors.

Siddiqui (2020), Maital and Barzani (2020), and Khan, Ullah, Usman, and Malik (2020) all provide comprehensive reviews of the global economic impact of COVID-19. They discuss the sharp contraction in global GDP, the financial market volatility, and the fiscal measures taken by governments worldwide to mitigate the economic damage.

In conclusion, the COVID-19 pandemic has had a profound and widespread impact on the global economy. The research presented here provides a detailed and informative overview of this impact, from immediate economic consequences to potential long-term changes in the global economic landscape (Bagchi et al., 2020; Naseer et al., 2023; Mishra, 2020; Priya et al., 2021; Shrestha et al., 2020; Siddiqui, 2020; Maital & Barzani, 2020; Khan et al., 2020).

Références:
1. Impact of COVID-19 on global economy - ['B Bagchi', 'S Chatterjee', 'R Ghosh', 'D Dandapat'] (2020)
2. COVID-19 outbreak: Impact on global economy - ['S Naseer', 'S Khalid', 'S Parveen', 'K Abbass'] (2023)
3. The World after COVID-19 and its impact on Global Economy - ['MK Mishra'] (2020)
4. The World Economy at COVID-19 quarantine: contemporary review - ['HL Feyisa'] (2020)
5. Covid-19 global, pandemic impact on world economy - ['MF Sattar', 'S Khanum', 'A Nawaz', 'MM Ashfaq'] (2020)
6. A perspective of COVID 19 impact on global economy, energy and environment - ['SS Priya', 'E Cuce', 'K Sudhakar'] (2021)
7. The impact of COVID-19 on globalization - ['N Shrestha', 'MY Shad', 'O Ulvi', 'MH Khan'] (2020)
8. The Impact of COVID-19 on the Global economy - ['K Siddiqui'] (2020)
9. The global economic impact of COVID-19: A summary of research - ['S Maital', 'E Barzani'] (2020)
10. Impact of covid 19 on the global economy - ['MA Khan', 'M Ullah', 'A Usman', 'HA Malik'] (2020)